flux tensoriel : : opérations : : BatchMatMul

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 .

Retours :

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

Attributs publics

Operation operation

::tensorflow::Output output

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AdjX(
  bool x
)

Attrs AdjY(
  bool x
)