python - 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)