如果a
是多维数组,则a[..., j]
给出形状为a.shape[:-1]
的数组。因此,例如,如果a
是m
乘n
的矩阵,则a[..., j]
给出形状为m
的{{1}}维向量。什么是进行上述切片的正确方法,以使结果具有与(m,)
相同的维数,而最后一个维的大小为a
。例如,1
乘a[<magical_slice>]
数组的m
应给出形状为n
的(列)向量。
当前我正在使用
(m, 1)
实现该目标。但是,对于我想做的事情来说,看起来太复杂了(而且很可能效率不高)。