Numpy `Broadcast` array

Question

I would like to do a transformation like dimshuffle in theano using numpy.

Example input:

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

Example output:

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

hpaulj · Accepted Answer · 2015-12-08 01:29:48Z

1

I don't know what dimshuffle does, but the output can be produced with 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]]])

The input is 2d (2,3), so I have to add an axis - output is (3,2,3). tile would work, so would indexing, or even:

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

answered Dec 8, 2015 at 1:29

hpaulj

233k14 gold badges260 silver badges392 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Numpy `Broadcast` array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related