torch.unsqueeze 对应numpy
时间: 2023-10-16 17:01:17 浏览: 106
Pyorch之numpy与torch之间相互转换方式
torch.unsqueeze 在 PyTorch 中的功能类似于 numpy 的 np.expand_dims。它用于在给定维度上增加一个维度。例如,如果有一个形状为 (3, 4) 的张量,使用 torch.unsqueeze(dim=0) 将会得到一个形状为 (1, 3, 4) 的张量,其中在第 0 维上增加了一个维度。这与 numpy 的 np.expand_dims(arr, axis=0) 的功能相同。
阅读全文