在 pytorch 中是否存在包含列表的张量的 tensor.item() 等价物？
Is there a tensor.item() equivalent for a tensor containing a list in pytorch?
在pytorch中，如果我定义一个单元素张量如下：
>>> import torch
>>> target1 = torch.tensor([5])
我可以像这样提取它的一个元素的值：
>>> target1.item()
5
我想知道的是当我的张量定义为：
target2 = torch.tensor([[5], [5], [5], [5]])
是否有某种方法（类似于或不类似于上面的 .item()）将其所有条目提取到列表中，例如：
>>> target2.(something)
[5, 5, 5, 5]
我似乎在文档中找不到任何支持此类操作的函数。
您可以使用
target2.numpy().ravel() 或
target2.view(-1).numpy() 或
target2.view(target2.numel()).numpy()
Out[1]: array([5, 5, 5, 5], dtype=int64)
在pytorch中，如果我定义一个单元素张量如下：
>>> import torch
>>> target1 = torch.tensor([5])
我可以像这样提取它的一个元素的值：
>>> target1.item()
5
我想知道的是当我的张量定义为：
target2 = torch.tensor([[5], [5], [5], [5]])
是否有某种方法（类似于或不类似于上面的 .item()）将其所有条目提取到列表中，例如：
>>> target2.(something)
[5, 5, 5, 5]
我似乎在文档中找不到任何支持此类操作的函数。
您可以使用
target2.numpy().ravel() 或
target2.view(-1).numpy() 或
target2.view(target2.numel()).numpy()
Out[1]: array([5, 5, 5, 5], dtype=int64)