从左往右看，torch.eq()是比较input和output的函数,input必须为tensor类型，output可以为相同大小的tensor也可以为某个值，当input和output都为tensor类型时，比较对应位置数字是否相同，相同则为1，否则为0。而torch.eq()函数得到的是个tensor。
例如：A = [1,2,3,4]
B = [1,1,2,2]
那么torch.eq(A,B)得到的结果就是[1,0,0,0]
torch.eq().sum()就是将所有值相加，但是得到的仍然是一个tensor,本例中torch.eq(A,B).sum()得到的结果就是[1](1+0+0+0),最后一步torch.eq(A,B).sum().item()得到的就是这个tensor中的值了，即1。
最后
以上就是结实大叔最近收集整理的关于torch.eq(input,output).sum().item()的全部内容，更多相关torch内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复