Einsum

cours final public Einsum

Une opération qui prend en charge une opération de base avec 2 entrées et 1 sortie.

Cette opération a de meilleures performances TPU car elle n'a pas d'opérations de remodelage et de transposition explicites comme le fait tf.einsum.

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique du tenseur.
static <T étend TType > Un seul <T>
créer ( Portée de portée , Opérande <T> a, Opérande <T> b, Équation de chaîne)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération Einsum.
Sortie <T>
produit ()

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 : "XlaEinsum"

Méthodes publiques

sortie publique <T> asOutput ()

Renvoie le handle symbolique du tenseur.

Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.

public static Einsum <T> create ( Portée de la portée , Opérande <T> a, Opérande <T> b, Équation de chaîne)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération Einsum.

Paramètres
portée portée actuelle
Retour
  • une nouvelle instance d'Einsum

public Sortie <T> produit ()