тензорный поток:: опс:: РазреженныйТензор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
)