tensorflow :: opérations :: SparseTensorDenseMatMul

#include <sparse_ops.h>

Multipliez SparseTensor (de rang 2) "A" par la matrice dense "B".

Résumé

Aucune vérification de validité n'est effectuée sur les indices de A. Cependant, le format d'entrée suivant est recommandé pour un comportement optimal:

if adjoint_a == false: A doit être trié dans un ordre lexicographique croissant. Utilisez SparseReorder si vous n'êtes pas sûr. si adjoint_a == true: A doit être trié par ordre croissant de dimension 1 (c'est-à-dire, ordre "colonne principale" au lieu de "ligne principale").

Arguments:

  • scope: un objet Scope
  • a_indices: 2-D. Les indices de la SparseTensor , taille [nnz, 2] .
  • a_values: 1-D. Les values du SparseTensor , size [nnz] Vector.
  • une_forme: 1-D. La shape du SparseTensor , taille [2] .
  • b: 2-D. Une matrice dense.

Attributs facultatifs (voir Attrs ):

  • adjoint_a: Utilise l'adjoint de A dans la matrice de multiplication. Si A est complexe, c'est transposer (conj (A)). Sinon, c'est transposer (A).
  • adjoint_b: Utilise l'adjoint de B dans la matrice multipliée. Si B est complexe, c'est transpose (conj (B)). Sinon, c'est transposer (B).

Retour:

  • Output : le tenseur du produit.

Constructeurs et destructeurs

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)

Attributs publics

operation
product

Fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Fonctions statiques publiques

AdjointA (bool x)
AdjointB (bool x)

Structs

tensorflow :: ops :: SparseTensorDenseMatMul :: Attrs

Définisseurs d' attributs facultatifs pour SparseTensorDenseMatMul .

Attributs publics

opération

Operation operation

produit

::tensorflow::Output product

Fonctions publiques

SparseTensorDenseMatMul

 SparseTensorDenseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input a_values,
  ::tensorflow::Input a_shape,
  ::tensorflow::Input b
)

SparseTensorDenseMatMul

 SparseTensorDenseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input a_values,
  ::tensorflow::Input a_shape,
  ::tensorflow::Input b,
  const SparseTensorDenseMatMul::Attrs & attrs
)

nœud

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const 

Fonctions statiques publiques

AdjointA

Attrs AdjointA(
  bool x
)

AdjointB

Attrs AdjointB(
  bool x
)