Умножьте матрицу «а» на матрицу «б».
Входные данные должны быть двумерными матрицами, а внутреннее измерение «a» должно соответствовать внешнему измерению «b». И «a», и «b» должны быть Tensor, а не SparseTensor. Эта операция оптимизирована для случая, когда хотя бы одно из «a» или «b» является разреженным, в том смысле, что они имеют большую долю нулевых значений. Безубыточность использования этого метода по сравнению с умножением плотной матрицы на одной платформе составила 30% нулевых значений в разреженной матрице.
Вычисление градиента этой операции будет использовать преимущества разреженности входного градиента только в том случае, если этот градиент исходит от Relu.
Вложенные классы
сорт | SparseMatMul.Options | Дополнительные атрибуты для SparseMatMul |
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
статический SparseMatMul.Options | aIsSparse (логическое значение aIsSparse) |
Вывод <TFloat32> | какВывод () Возвращает символический дескриптор тензора. |
статический SparseMatMul.Options | bIsSparse (логическое значение bIsSparse) |
статический SparseMatMul | create ( Область действия , Операнд <? расширяет TNumber > a, Операнд <? расширяет TNumber > b, Параметры... параметры) Фабричный метод для создания класса, обертывающего новую операцию SparseMatMul. |
Вывод <TFloat32> | продукт () |
статический SparseMatMul.Options | transposeA (логическое значение transposeA) |
статический SparseMatMul.Options | transposeB (логическое значение transposeB) |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <TFloat32> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static SparseMatMul create (область действия , операнд <? расширяет TNumber > a, операнд <? расширяет TNumber > b, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию SparseMatMul.
Параметры
объем | текущий объем |
---|---|
параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр SparseMatMul