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方式(逐元素操作的方式),此时要求两个数据的维度必须相同。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 单元测试从入门到精通
· 上周热点回顾(3.3-3.9)
· Vue3状态管理终极指南:Pinia保姆级教程