火炬张量添加维度

Question

我有一个这种大小的张量

torch.Size([128, 64])

如何添加一个“虚拟”维度，例如

torch.Size([1, 128, 64])

Answer 1

有几种方法：

torch.unsqueeze：

torch.unsqueeze(x, 0)

---

使用None（或np.newaxis）：

x[None, ...]
# or
x[np.newaxis, ...]

---

reshape 或 view：

x.reshape(1, *x.shape)
# or
x.view(1, *x.shape)