pytorch 判断两个 tensor 是否相等

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

import torch

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

torch.equal(x, y)

你可能感兴趣的:(pytorch 判断两个 tensor 是否相等)