概述
从左往右看,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.eq(input,output).sum().item()所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复