MXnet基础(1)

创建变量

from mxnet import nd
#NDArray 类似于Tensor 基本和np的使用方式一致
    [[2 1 4 6]
     [1 4 3 4]
     [4 6 2 1]]
    <NDArray 3x4 @cpu(0)>
nd.array([])
nd.arrange(n)
nd.ones(shape)
nd.zeros()
nd.reshape(shape)
nd.random.rand #Series

 

变量操作

• sum
• concat
• dot (mxnet只支持float的矩阵乘法)
• norm
• asscalar
• asnumpy
• squeeze

  s=nd.arange(10).reshape((1,2,5,1,1))
  print(s)
  print(nd.squeeze(s))
  [[[[[0.]]
     [[1.]]
     [[2.]]
     [[3.]]
     [[4.]]]
    [[[5.]]
     [[6.]]
     [[7.]]
     [[8.]]
     [[9.]]]]]
  <NDArray 1x2x5x1x1 @cpu(0)>
  
  [[0. 1. 2. 3. 4.]
   [5. 6. 7. 8. 9.]]
  <NDArray 2x5 @cpu(0)>

   

•  

索引

索引基本和切片一样 以行为index (参考pandas)

Y[1:3] #取1,2行
Y[1:3,2]#取1,2行