Умножьте матрицу «а» на матрицу «б».
Входные данные должны быть двумерными матрицами, а внутреннее измерение «a» (после транспонирования, если transpose_a истинно) должно соответствовать внешнему измерению «b» (после транспонирования, если transpose_b истинно).
Примечание . Реализация ядра по умолчанию для MatMul на графических процессорах использует cublas.
Вложенные классы
| сорт | МатМул.Параметры | Дополнительные атрибуты для MatMul | |
Константы
| Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
| Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
| статический <T расширяет TType > MatMul <T> | create ( Область действия , Операнд <T> a, Операнд <T> b, Параметры... параметры) Фабричный метод для создания класса, обертывающего новую операцию MatMul. |
| Выход <Т> | продукт () |
| статический MatMul.Options | transposeA (логическое значение transposeA) |
| статический MatMul.Options | transposeB (логическое значение transposeB) |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static MatMul <T> create (область действия , операнд <T> a, операнд <T> b, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию MatMul.
Параметры
| объем | текущий объем |
|---|---|
| параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр MatMul
общедоступный статический MatMul.Options transposeA (логическое значение transposeA)
Параметры
| транспонироватьА | Если это правда, «a» транспонируется перед умножением. |
|---|
общедоступный статический MatMul.Options transposeB (логическое значение transposeB)
Параметры
| транспонировать Б | Если это правда, «b» транспонируется перед умножением. |
|---|