火炬张量添加维度

Question

我有一个这样大小的张量

return [activity for activity in values if activity["action"] == "COMMENTED"]

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

torch.Size([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)