易学智能论坛 › PyTorch › pytorch下只打印tensor的数值不打印出device等信息的方法

shaoheshaohe 发表于 2019-11-6 00:13:00

pytorch下只打印tensor的数值不打印出device等信息的方法

torch.Tensor类型的数据loss和acc打印时 ，如果写成以下写法

print('batch_loss: '+str(loss.data)+'batch acc: '+str(acc.data))
1
则不仅会打印出loss和acc的值，还会打印出device信息和 tensor字样，如下：

如果仅想打印出数值，使得打印出的信息更加简洁，则要用以下写法

print('batch_loss: {:.3f} batch acc: {:.3f}'.format(loss.data, acc.data))

shaoheshaohe 发表于 2019-11-6 00:13:07

https://blog.csdn.net/sjtuxx_lee/article/details/83116427

查看完整版本: pytorch下只打印tensor的数值不打印出device等信息的方法