通过 .item()获取Tensor的值,输出为numpy格式, 仅能得到维度是1 的数值
如
tensor.item()
通过.data 获取Tensor的值
tensor.data
tensor.data.cpu().numpy()
通过 .item()获取Tensor的值,输出为numpy格式, 仅能得到维度是1 的数值
如
tensor.item()
通过.data 获取Tensor的值
tensor.data
tensor.data.cpu().numpy()