тензорный поток:: опс:: РазреженныйТензорDenseMatMul
#include <sparse_ops.h>
Умножьте SparseTensor (ранга 2) «A» на плотную матрицу «B».
Краткое содержание
Для индексов A проверка достоверности не выполняется. Однако для оптимального поведения рекомендуется следующий формат ввода:
if adjoint_a == false: A следует отсортировать в лексикографическом порядке возрастания. Если вы не уверены, используйте SparseReorder . if adjoint_a == true: A следует отсортировать в порядке возрастания измерения 1 (т. е. в порядке «основной столбец» вместо порядка «основной ряд»).
Аргументы:
- область: объект области.
- a_indices: 2-D.
indices
матрицыSparseTensor
размера[nnz, 2]
. - a_values: 1-D.
values
SparseTensor
, размер[nnz]
Vector. - a_shape: 1-D.
shape
SparseTensor
, размер[2]
Вектор. - б: 2-Д. Плотная Матрица.
Необязательные атрибуты (см. Attrs
):
- adjoint_a: использовать сопряженное число A при умножении матрицы. Если A комплексное, это транспонирование (conj(A)). В противном случае это транспонирование (A).
- adjoint_b: использовать сопряженное к B при умножении матрицы. Если B комплексный, это транспонирование (conj(B)). В противном случае это транспонирование (B).
Возврат:
-
Output
: Тензор произведения.
Конструкторы и деструкторы | |
---|---|
SparseTensorDenseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b) | |
SparseTensorDenseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b, const SparseTensorDenseMatMul::Attrs & attrs) |
Публичные атрибуты | |
---|---|
operation | |
product |
Общественные функции | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Публичные статические функции | |
---|---|
AdjointA (bool x) | |
AdjointB (bool x) |
Структуры | |
---|---|
tensorflow:: ops:: SparseTensorDenseMatMul:: Attrs | Необязательные установщики атрибутов для SparseTensorDenseMatMul . |
Публичные атрибуты
операция
Operation operation
продукт
::tensorflow::Output product
Общественные функции
РазреженныйТензорDenseMatMul
SparseTensorDenseMatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a_indices, ::tensorflow::Input a_values, ::tensorflow::Input a_shape, ::tensorflow::Input b )
РазреженныйТензорDenseMatMul
SparseTensorDenseMatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a_indices, ::tensorflow::Input a_values, ::tensorflow::Input a_shape, ::tensorflow::Input b, const SparseTensorDenseMatMul::Attrs & attrs )
узел
::tensorflow::Node * node() const
оператор::tensorflow::Input
operator::tensorflow::Input() const
оператор::tensorflow::Выход
operator::tensorflow::Output() const
Публичные статические функции
АдджоинтА
Attrs AdjointA( bool x )
Примыкающий Б
Attrs AdjointB( bool x )