Skip to content

Latest commit

 

History

History
66 lines (37 loc) · 1.16 KB

mm_cn.rst

File metadata and controls

66 lines (37 loc) · 1.16 KB
Raw

mm

.. py:function:: paddle.mm(input, mat2, name=None)

用于两个输入矩阵的相乘。

两个输入的形状可为任意维度，但当任一输入维度大于 3 时，两个输入的维度必须相等。

如果原始 Tensor input 或 mat2 的秩为 1 且未转置，则矩阵相乘后的前置或附加维度 1 将移除。

参数

  • input (Tensor)：输入变量，类型为 Tensor。
  • mat2 (Tensor)：输入变量，类型为 Tensor。
  • name (str，可选) - 具体用法请参见 :ref:`api_guide_Name`，一般无需设置，默认值为 None。

返回

  • Tensor，矩阵相乘后的结果。
* 例 1:

input: [B, ..., M, K], mat2: [B, ..., K, N]
out: [B, ..., M, N]

* 例 2:

input: [B, M, K], mat2: [B, K, N]
out: [B, M, N]

* 例 3:

input: [B, M, K], mat2: [K, N]
out: [B, M, N]

* 例 4:

input: [M, K], mat2: [K, N]
out: [M, N]

* 例 5:

input: [B, M, K], mat2: [K]
out: [B, M]

* 例 6:

input: [K], mat2: [K]
out: [1]

代码示例

COPY-FROM: paddle.mm