我是靠谱客的博主 结实大叔,最近开发中收集的这篇文章主要介绍torch.eq(input,output).sum().item(),觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

从左往右看,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()所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(38)

评论列表共有 0 条评论

立即
投稿
返回
顶部