reshape函数 0 -1 的具体含义

reshape((0,-1,N))

  • 0:保持原有的通道数不变
  • -1:自动计算通道数值。根据剩下的维度计算出数组的另外一个shape属性值。
a = nd.random.randint(1,50,shape=(4,3,3))
a

 输出结果:

[[[48 25 22]
  [ 5 49  6]
  [34 19 23]]

 [[33 28 13]
  [35  9  8]
  [29 32 47]]

 [[40  3 14]
  [40 32  7]
  [ 6 49 12]]

 [[32 43  8]
  [26 25 36]
  [ 9 45  2]]]

使用reshape 修改形状

a.reshape(0,-1,4)

输出结果: 

[[[48 25 22  5]
  [49  6 34 19]]

 [[23 33 28 13]
  [35  9  8 29]]

 [[32 47 40  3]
  [14 40 32  7]]

 [[ 6 49 12 32]
  [43  8 26 25]]]

你可能感兴趣的:(深度学习,机器学习)