BroadcastHelper

classe finale publique BroadcastHelper

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>
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.
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

Valeur constante : "XlaBroadcastHelper"

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

sortie publique <T> lhsOutput ()

le tenseur LHS diffusé

sortie publique <T> rhsOutput ()

le tenseur RHS diffusé