tips

1.nn.EmbeddingBag:适合用于文本进行ngram特征表示,有sum,mean等方式

 2. dot,*,mul,@,matmul,mm,bmm

torch.dot(a,b)

两向量相乘相加得到一个标量,必须都是一维的

torch.bmm(a,b):矩阵乘法

批矩阵相乘,不会进行广播,必须满足矩阵相乘维数条件,a,b最多只能3维,且a,b中必须包含相同的矩阵个数即a,b第一维度必须相同

torch.mm():矩阵乘法等同于@

矩阵相乘,不会进行广播,必须满足矩阵相乘维数条件,两矩阵最多是2维

*和mul是element-wise方式(逐元素操作的方式),此时要求两个数据的维度必须相同。

posted @   15375357604  阅读(5)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 单元测试从入门到精通
· 上周热点回顾(3.3-3.9)
· Vue3状态管理终极指南:Pinia保姆级教程
点击右上角即可分享
微信分享提示