列表張量：如何轉換為一個列表的張量？

Question

我有這樣的張量

tensor([[4.],[1.]], device='cuda:0')

我想更改為以下內容：

tensor([4.,1.], device='cuda:0')

怎么做？

我不確定這是否是錯誤的原因：

loss = nn.MSELoss(y_tilde,y)

RuntimeError: Boolean value of Tensor with more than one value is ambiguous

Answer 1

使用

torch.flatten()

例子：

t = torch.tensor([[[1, 2],
                   [3, 4]],
                  [[5, 6],
                 [7, 8]]])
out: tensor([[[1, 2],
         [3, 4]],

        [[5, 6],
         [7, 8]]])
torch.flatten(t)
out: tensor([1, 2, 3, 4, 5, 6, 7, 8])