torch.Tensor的四种乘法操作具体是哪四种？

本文共计1524个文字，预计阅读时间需要7分钟。

`torch.Tensor` 提供四种常见乘法操作：`*`, `torch.mul`, `torch.mm`, `torch.matmul`。简要说明这四种乘法的区别，具体使用还需参照官方文档。

1. `*`：点乘，适用于元素级运算，当两个张量的维度相同时使用。

2.`torch.mul`：元素级乘法，与 `*` 功能相同。

3.`torch.mm`：矩阵乘法，适用于二维张量，即矩阵与矩阵或矩阵与向量的乘法。

4.`torch.matmul`：矩阵乘法，支持不同维度的张量，功能更强大。

例如，点乘 `a * b` 的原则是当 `a` 和 `b` 的 size 相同时，对应元素相乘。

torch.Tensor有4种常见的乘法：*, torch.mul, torch.mm, torch.matmul. 本文抛砖引玉，简单叙述一下这4种乘法的区别，具体使用还是要参照官方文档。

点乘

a与b做*乘法，原则是如果a与b的size不同，则以某种方式将a或b进行复制，使得复制后的a和b的size相同，然后再将a和b做element-wise的乘法。

下面以*标量和*一维向量为例展示上述过程。