mxnet slice_axis

 

mx.symbol.slice_axis可以直接在某一维上切割，选择整行或整列。

pick 是精准挑选指定位置的值，

 

mxnet slice_axis

begin end参数， 左开右闭，左边包含，右边不包含

 

import mxnet as mx
a = mx.nd.arange(18).reshape((3,2,3))

print(a.shape)
print(a)

d = mx.nd.slice_axis(a, axis=0, begin=0, end=2)
data=d.asnumpy()
print(data)