pytorch tensor整数除法

发表于 | 阅读次数: 
1
2
3
4
a = torch.Tensor([1, 3])
b = torch.Tensor([1, 4])
torch.sum(a == b) * 1.0 / 3 # ==> 0
torch.sum(a == b).item() * 1.0 / 3
请作者喝一杯咖啡☕️