Numpy ` `Broadcast` `数组

Question

我想做一个像theano中的dimshuffle使用numpy的转换。

示例输入：

np.array([[1, 0, 0], [1, 0, 0]])

输出示例：

np.array([
    [[1, 0, 0], [1, 0, 0]],
    [[1, 0, 0], [1, 0, 0]],
    [[1, 0, 0], [1, 0, 0]]
])

Answer 1

我不知道dimshuffle是做什么的，但是可以用repeat生成输出

In [319]: np.repeat(np.array([[1, 0, 0], [1, 0, 0]])[None,:,:],3,axis=0)
Out[319]: 
array([[[1, 0, 0],
        [1, 0, 0]],

       [[1, 0, 0],
        [1, 0, 0]],

       [[1, 0, 0],
        [1, 0, 0]]])

输入是2d轴，所以我必须添加一个轴- (2,3) is (3,2,3)。tile会起作用，索引也会起作用，甚至：

A=np.array([[1, 0, 0], [1, 0, 0]])
np.array([A,A,A])