tensorflow :: opérations :: BatchMatMulV2

Multiplie les tranches de deux tenseurs par lots.

Résumé

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

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

Le tenseur de sortie est 2-D ou plus 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 plus avec forme [..., r_x, c_x] .
• y: 2D ou plus avec forme [..., r_y, c_y] .

Attributs facultatifs (voir Attrs ):

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

Retour:

• Output : 3-D ou plus avec 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
)