tf.keras.ops.matmul | TensorFlow v2.16.1 (original) (raw)
tf.keras.ops.matmul
Matrix product of two tensors.
View aliases
Main aliases
tf.keras.ops.matmul(
x1, x2
)
- If both tensors are 1-dimensional, the dot product (scalar) is returned.
- If either tensor is N-D, N > 2, it is treated as a stack of matrices residing in the last two indexes and broadcast accordingly.
- If the first tensor is 1-D, it is promoted to a matrix by prepending a 1 to its dimensions. After matrix multiplication the prepended 1 is removed.
- If the second tensor is 1-D, it is promoted to a matrix by appending a 1 to its dimensions. After matrix multiplication the appended 1 is removed.
| Args | |
|---|---|
| x1 | First tensor. |
| x2 | Second tensor. |
| Returns |
|---|
| Output tensor, matrix product of the inputs. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-06-07 UTC.