Spec-Zone.ru › TensorFlow C++ 2.9

tensorflow::ops::SparseMatMul

#include <math_ops.h>

Multiply матрицу "a" на матрицу "b".

Краткое описание

Входы должны быть двумерными матрицами, а внутренняя размерность "a" должна совпадать с внешней размерностью "b". Обе "a" и "b" должны быть Tensors, а не SparseTensors. Этот оператор оптимизирован для случая, когда хотя бы одна из "a" или "b" является разреженной, в том смысле, что у них большая доля нулевых значений. Критерий перехода от использования этого к умножению плотных матриц на одной платформе составил 30% нулевых значений в разреженной матрице.

Вычисление градиента этой операции будет использовать разреженность входного градиента только в том случае, если этот градиент получен от Relu.

Аргументы:

  • scope: Объект Scope

Возвращает:

  • Output: Тензор произведения.
Конструкторы и деструкторы
SparseMatMul(const ::tensorflow::Scope & scope, ::tensorflow::Input a, ::tensorflow::Input b)
SparseMatMul(const ::tensorflow::Scope & scope, ::tensorflow::Input a, ::tensorflow::Input b, const SparseMatMul::Attrs & attrs)
Общедоступные атрибуты
operation
Operation
product
::tensorflow::Output
Общедоступные функции
node() const
::tensorflow::Node *
operator::tensorflow::Input() const
operator::tensorflow::Output() const
Общедоступные статические функции
AIsSparse(bool x)
Attrs
BIsSparse(bool x)
Attrs
TransposeA(bool x)
Attrs
TransposeB(bool x)
Attrs
Структуры
tensorflow::ops::SparseMatMul::Attrs

Дополнительные атрибуты для установки для SparseMatMul.

Общедоступные атрибуты

operation

Operation operation

product

::tensorflow::Output product

Общедоступные функции

SparseMatMul

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b
)

SparseMatMul

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b,
  const SparseMatMul::Attrs & attrs
)

узел

::tensorflow::Node * node() const 

оператор::tensorflow::Input

operator::tensorflow::Input() const 

оператор::tensorflow::Output

operator::tensorflow::Output() const 

Общедоступные статические функции

AIsSparse

Attrs AIsSparse(
  bool x
)

BIsSparse

Attrs BIsSparse(
  bool x
)

TransposeA

Attrs TransposeA(
  bool x
)

TransposeB

Attrs TransposeB(
  bool x
)

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/cc/class/tensorflow/ops/sparse-mat-mul

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API