tensorflow::ops::MatMul
#include <math_ops.h>
Умножает матрицу «a» на матрицу «b».
Сводка
Входные данные должны быть двумерными матрицами, а внутренняя размерность «a» (после транспонирования, если transpose_a имеет значение true) должна совпадать с внешней размерностью «b» (после транспонирования, если transpose_b имеет значение true).
Примечание: Реализация ядра по умолчанию для MatMul на GPU использует cublas.
Аргументы:
- scope: объект Scope
Необязательные атрибуты (см. Attrs):
- transpose_a: если true, матрица «a» транспонируется перед умножением.
- transpose_b: если true, матрица «b» транспонируется перед умножением.
Возвращает:
-
Output: тензор-произведение.
| Конструкторы и деструкторы | |
|---|---|
MatMul(const ::tensorflow::Scope & scope, ::tensorflow::Input a, ::tensorflow::Input b) | |
MatMul(const ::tensorflow::Scope & scope, ::tensorflow::Input a, ::tensorflow::Input b, const MatMul::Attrs & attrs) |
| Публичные атрибуты | |
|---|---|
operation | |
product | |
| Публичные функции | |
|---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const | |
| Публичные статические функции | |
|---|---|
TransposeA(bool x) | |
TransposeB(bool x) | |
| Структуры | |
|---|---|
| tensorflow::ops::MatMul::Attrs | Методы установки необязательных атрибутов для MatMul. |
Публичные атрибуты
operation
Operation operation
product
::tensorflow::Output product
Публичные функции
MatMul
MatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a, ::tensorflow::Input b )
MatMul
MatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a, ::tensorflow::Input b, const MatMul::Attrs & attrs )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Публичные статические функции
TransposeA
Attrs TransposeA( bool x )
TransposeB
Attrs TransposeB( bool x )
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/cc/class/tensorflow/ops/mat-mul