Pytorch: RuntimeError: 零维张量（在位置 0）不能连接

Question

我有两个张量

# losses_q
tensor(0.0870, device='cuda:0', grad_fn=<SumBackward0>)
# this_loss_q
tensor([0.0874], device='cuda:0', grad_fn=<AddBackward0>)

当我尝试连接它们时，pytorch 引发错误：

losses_q = torch.cat((losses_q, this_loss_q), dim=0)

RuntimeError: zero-dimensional tensor (at position 0) cannot be concatenated

如何解决这个错误？

Answer 1

losses_q

是零维的，所以不能与任何东西连接。您可以在连接之前将其重塑为一维张量。

losses_q = torch.cat((losses_q.reshape(1), this_loss_q), dim=0)