test_mxnet_function



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 [ ]: