flux tensoriel : : opérations : : BatchMatMulV2

#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[..., :, :])

REMARQUE : BatchMatMulV2 prend en charge la diffusion dans les dimensions du lot. En savoir plus sur la diffusion ici .

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 .

Retours :

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

Constructeurs et Destructeurs

BatchMatMulV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y)
BatchMatMulV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y, const BatchMatMulV2::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 :: BatchMatMulV2 :: Attrs

Setters d'attributs facultatifs pour BatchMatMulV2 .

Attributs publics

opération 

Operation operation

sortir 

::tensorflow::Output output

Fonctions publiques

BatchMatMulV2 

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

BatchMatMulV2 

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y,
  const BatchMatMulV2::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
)