tensorflow::ops::BatchMatMulV2

Multiplie des tranches de deux tenseurs par lots.

Résumé

Cette fonction multiplie toutes les tranches des Tensor x et y (chaque tranche pouvant être considérée comme un élément d'un lot) et organise les résultats individuels en un seul tenseur de sortie de même taille de lot. Chaque tranche peut être adjointe (l'adjonction d'une matrice consiste à la transposer et à la conjuguer) avant la multiplication en définissant l'indicateur adj_x ou adj_y sur True , qui est par défaut False .

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

Le tenseur de sortie est de dimension 2D ou supérieure et de 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. Plus d’informations sur la diffusion ici .

Arguments :

• portée : un objet Scope
• x : 2D ou plus avec la forme [..., r_x, c_x] .
• y : 2D ou plus avec la forme [..., r_y, c_y] .

Attributs optionnels (voir Attrs ) :

• adj_x : Si True , adjoint aux tranches de x . Par défaut : False .
• adj_y : Si True , adjoint aux tranches de y . Par défaut : False .

Retours :

• Output : 3D ou plus avec la forme [..., r_o, c_o]

attributs publics

Operation operation

::tensorflow::Output output

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AdjX(
  bool x
)

Attrs AdjY(
  bool x
)