tensorflow :: opérations :: SparseTensorDenseMatMul

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.

Attributs publics

Operation operation

::tensorflow::Output product

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AdjointA(
  bool x
)

Attrs AdjointB(
  bool x
)