flux tensoriel : : opérations : : BatchMatMul

#include <math_ops.h>

Multiplie les tranches de deux tenseurs par lots.

Résumé

Multiplie toutes les tranches du Tensor x et y (chaque tranche peut être considérée comme un élément d'un lot) et organise les résultats individuels dans un seul tenseur de sortie de la même taille de lot. Chacune des tranches individuelles peut éventuellement être adjointe (pour adjoint une matrice signifie la transposer et la conjuguer) avant la multiplication en définissant l'indicateur adj_x ou adj_y sur True , qui sont par défaut False .

Les tenseurs d'entrée x et y sont 2D ou supérieurs avec la forme [..., r_x, c_x] et [..., r_y, c_y] .

Le tenseur de sortie est 2D ou supérieur avec la forme [..., r_o, c_o] , où :

r_o = c_x if adj_x else r_x
c_o = r_y if adj_y else c_y

Il est calculé comme suit :

output[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])

Arguments:

  • scope : un objet Scope
  • x : 2D ou supérieur avec la forme [..., r_x, c_x] .
  • y : 2D ou supérieur avec la forme [..., r_y, c_y] .

Attributs facultatifs (voir Attrs ) :

  • adj_x : Si True , adjoint les tranches de x . La valeur par défaut est False .
  • adj_y : Si True , adjoint les tranches de y . La valeur par défaut est False .

Retour:

  • Output : 3D ou supérieure avec forme [..., r_o, c_o]

Constructeurs et Destructeurs

BatchMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y)
BatchMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y, const BatchMatMul::Attrs & attrs)

Attributs publics

operation
output

Fonctions publiques

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

Fonctions statiques publiques

AdjX (bool x)
AdjY (bool x)

Structures

tensorflow :: ops :: BatchMatMul :: Attrs

Setters d'attributs facultatifs pour BatchMatMul .

Attributs publics

opération

Operation operation

sortir

::tensorflow::Output output

Fonctions publiques

BatchMatMul

 BatchMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y
)

BatchMatMul

 BatchMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y,
  const BatchMatMul::Attrs & attrs
)

nœud

::tensorflow::Node * node() const 

opérateur :: tensorflow :: Entrée

 operator::tensorflow::Input() const 

opérateur :: tensorflow :: Sortie

 operator::tensorflow::Output() const 

Fonctions statiques publiques

AjX

Attrs AdjX(
  bool x
)

AdjY

Attrs AdjY(
  bool x
)