首页 > 解决方案 > 如何使用 pytorch 使这种乘法变得更简单?

问题描述

我想像这样将两个矩阵相乘(标量乘法)。但这种方法看起来很愚蠢。我怎样才能让这更简单?

A.size # B * S * V
B.size # B

A * B.unsqueeze(1).unsqueeze(2).expand_as(A)

标签: pytorch

解决方案


推荐阅读