0
点赞
收藏
分享

微信扫一扫

torch tensor == return False

那小那小 2022-02-03 阅读 161
pytorch

在这里插入图片描述
输出a,b; 看上去两者都是0.5904。

在这里插入图片描述
但是两者却不相等,这是因为两者的item不一样
在这里插入图片描述

举报

相关推荐

torch.topk和tensor.topk

0 条评论