美文网首页
pytorch 判断两个 tensor 是否相等

pytorch 判断两个 tensor 是否相等

作者: 潘旭 | 来源:发表于2020-10-11 21:09 被阅读0次

不用循环,用 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()

相关文章

网友评论

      本文标题:pytorch 判断两个 tensor 是否相等

      本文链接:https://www.haomeiwen.com/subject/tydeuktx.html