未验证 提交 327f635e 编写于 作者: W wawltor 提交者: GitHub

add the examples for the mm (#38746)

上级 52679889
...@@ -932,6 +932,37 @@ def mm(input, mat2, name=None): ...@@ -932,6 +932,37 @@ def mm(input, mat2, name=None):
Returns: Returns:
Tensor: The product Tensor. Tensor: The product Tensor.
::
* example 1:
input: [B, ..., M, K], mat2: [B, ..., K, N]
out: [B, ..., M, N]
* example 2:
input: [B, M, K], mat2: [B, K, N]
out: [B, M, N]
* example 3:
input: [B, M, K], mat2: [K, N]
out: [B, M, N]
* example 4:
input: [M, K], mat2: [K, N]
out: [M, N]
* example 5:
input: [B, M, K], mat2: [K]
out: [B, M]
* example 6:
input: [K], mat2: [K]
out: [1]
Examples: Examples:
.. code-block:: python .. code-block:: python
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册