网站首页 站内搜索

搜索结果

查询Tags标签: bmm,共有 1条记录
  • pytorch 记录

    torch.take() 把张量化为一个行向量,取其中的元素 广播机制中0维和1维的轴是可广播的 矩阵相乘 *和torch.mul()是对应位置相乘 @和torch.mm()是矩阵相乘 torch.bmm()相比mm增加了batch运算但不能广播 torch.matmul()相比bmm更是增加了广播,泛用性强 torch.full([3,3],1…

    2022/4/12 23:44:58 人评论 次浏览
扫一扫关注最新编程教程