使用torch.squeeze函数将输入标签和预测标签裁剪为二维矩阵或者一维矩阵,当直接输入二维矩阵以上时,返回值就为0
torch.nn.CrossEntropyLoss返回值为0
最新推荐文章于 2024-03-05 11:37:00 发布
使用torch.squeeze函数将输入标签和预测标签裁剪为二维矩阵或者一维矩阵,当直接输入二维矩阵以上时,返回值就为0