pytorch 判断两个 tensor 是否相等

不用循环，用 pytorch 的基本函数, 非常简洁. 代码如下:

import torch

x = torch.tensor([[1, 2], [3, 4]])
y = torch.tensor([[1, 2], [3, 4]])

assert 0 == ((x != y).sum())

xx = torch.tensor([[1, 2], [3, 4]])
yy = torch.tensor([[2, 2], [3, 3]])
assert 0 == (xx != yy).sum()

承接Matlab、Python和C++的编程，机器学习、计算机视觉的理论实现及辅导，本科和硕士的均可，咸鱼交易，专业回答请走知乎，详谈请联系QQ号757160542，非诚勿扰。