仅作为记录,大佬请跳过。
用outputs[0][0]
outputs
# tensor([[ 7.7213, -7.2519]], device='cuda:0')
# tensor([[ 7.3663, -7.1830]], device='cuda:0')
# tensor([[-5.6146, 5.7881]], device='cuda:0')
# a=np.array((torch.max(outputs).data.cpu()-(-10))/20)
# a
outputs.size()
outputs[0][0]
参考