Умножьте матрицу «а» на матрицу «б».
Входные данные должны быть двумерными матрицами, а внутреннее измерение «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) |
Унаследованные методы
логическое значение | равно (Объект arg0) |
последний класс<?> | получитьКласс () |
интервал | хэш-код () |
окончательная пустота | поставить в известность () |
окончательная пустота | уведомитьВсе () |
Нить | нанизывать () |
окончательная пустота | подождать (длинный arg0, int arg1) |
окончательная пустота | подождите (длинный arg0) |
окончательная пустота | ждать () |
абстрактная среда выполнения | окружение () Верните среду выполнения, в которой была создана эта операция. |
абстрактная операция |
Константы
общедоступная статическая финальная строка 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» транспонируется перед умножением. |
---|