Opérateur assistant pour effectuer des diffusions de style XLA
Diffuse `lhs` et `rhs` au même rang, en ajoutant des dimensions de taille 1 à celui de `lhs` et `rhs` ayant le rang inférieur, en utilisant les règles de diffusion de XLA pour les opérateurs binaires.
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
statique <T étend TType > BroadcastHelper <T> | |
Sortie <T> | lhsSortie () le tenseur LHS diffusé |
Sortie <T> | rhsSortie () le tenseur RHS diffusé |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
public static BroadcastHelper <T> créer ( Scope scope, Operand <T> lhs, Operand <T> rhs, Operand <? extends TNumber > BroadcastDims)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération BroadcastHelper.
Paramètres
portée | portée actuelle |
---|---|
lhs | le tenseur d'entrée LHS |
rhs | le tenseur d'entrée RHS |
diffusionDims | une spécification de dimension de diffusion de style XLA |
Retour
- une nouvelle instance de BroadcastHelper