Python: None含义

None的作用

None的作用主要是在使用None的位置新增一个维度。

x[None] 等价于 x[None, :]

在最前面加一维

import torch
import numpy as np

a = np.arange(8).reshape(2,4)
print(a)
print(a.shape)
print("")

print(a[:,None])
print(a[:,None].shape)
print("")

print(a[:,:,None])
print(a[:,:,None].shape)

print(a[None])
print(a[None].shape)
print("")


输出：
[[0 1 2 3]
 [4 5 6 7]]
(2, 4)

[[[0 1 2 3]]

 [[4 5 6 7]]]
(2, 1, 4)

[[[0]
  [1]
  [2]
  [3]]

 [[4]
  [5]
  [6]
  [7]]]
(2, 4, 1)

[[[0 1 2 3]
  [4 5 6 7]]]
(1, 2, 4)