在pytorch中包含列表的张量是否有tensor.item（）等效项？

Question

在pytorch中，如果我按如下方式定义一元张量：

>>> import torch
>>> target1 = torch.tensor([5])

我能够像这样提取其一个元素的值：

>>> target1.item()
5

我想知道的是何时将我的张量定义为：

target2 = torch.tensor([[5], [5], [5], [5]])

是否可以通过某种方式（类似于或不同于上述 .item（））将其所有条目拉出到列表中，例如：

>>> target2.(something)
[5, 5, 5, 5]

我似乎在文档中找不到支持这种操作的任何功能。

Answer 1

您可以使用

target2.numpy().ravel()或

target2.view(-1).numpy()或

target2.view(target2.numel()).numpy()

Out[1]: array([5, 5, 5, 5], dtype=int64)