Spec-Zone.ru › TensorFlow C++

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
)

node

::tensorflow::Node * node() const 

operator::tensorflow::Input

operator::tensorflow::Input() const 

operator::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/api_docs/cc/class/tensorflow/ops/sparse-mat-mul

Spec-Zone.ru

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