- 博客(1)
- 收藏
- 关注
原创 【Pytorch警告】Using a target size (torch.Size([])) that is different to the input size (torch.Size([1])
【Pytorch警告】UserWarning: Using a target size (torch.Size([])) that is different to the input size (torch.Size([1])).【原因】mse_loss损失函数的两个输入Tensor的shape不一致。经过reshape或者一些矩阵运算以后使得shape一致,不再出现警告了。
2023-04-04 20:39:27 6264 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人