In [22]:

    

import mxnet as mx
from mxnet import nd


    

In [26]:

    

w = nd.ones(shape=(1,6,6,32,8,10,16))
x = nd.ones(shape=(1,6,6,32,8,1,1))
b = nd.ones(shape=(1,6,6,32,1,10,1))


    

In [27]:

    

y = x*w
y.shape


    

    
Out[27]:





(1, 6, 6, 32, 8, 10, 16)

In [29]:

    

z = x*b
z.shape


    

    
Out[29]:





(1, 6, 6, 32, 8, 10, 1)

In [30]:

    

z.argmax


    

    




---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-30-a9901bab8963> in <module>()
----> 1 z.argmax

AttributeError: 'NDArray' object has no attribute 'argmax'

In [ ]: