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()本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2021/03/26 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
腾讯云开发者
