TBfloat16

interface publique TBfloat16

Type de tenseur flottant cérébral 16 bits.

Ce type diffère de TFloat16 car il tronque la mantisse d'un flottant 32 bits et préserve tous les bits d'exposant pour une conversion plus rapide, tandis que ce dernier réduit l'exposant et a une mantisse plus longue pour plus de précision.

Puisqu'il n'existe pas de type à virgule flottante pouvant tenir sur 16 bits en Java, une conversion (avec potentiellement une perte de précision) est requise pour chaque valeur de 32 bits écrite ou lue sur un tenseur de ce type depuis la JVM. Par conséquent, si de nombreuses opérations d’E/S sont à prévoir sur un tenseur, les performances seront améliorées en travaillant avec les types de données TFloat32 ou TFloat64 autant que possible.

Notez que certains processeurs prennent en charge le format bfloat16 de manière native, ce qui peut entraîner un calcul plus rapide par rapport à TFloat16 lorsque les GPU ne sont pas utilisés.

Méthodes publiques

abstrait statique TBfloat16
scalarOf (valeur flottante)
Alloue un nouveau tenseur pour stocker une seule valeur flottante.
abstrait statique TBfloat16
tensorOf ( Forme , Consommateur < TBfloat16 > dataInit)
Alloue un nouveau tenseur de la forme donnée et initialise ses données.
abstrait statique TBfloat16
tensorOf ( Forme forme)
Alloue un nouveau tenseur de la forme donnée.
abstrait statique TBfloat16
tensorOf (forme de forme , données FloatDataBuffer )
Alloue un nouveau tenseur de la forme donnée, initialisé avec les données fournies.
abstrait statique TBfloat16
tensorOf ( NdArray <Float> src)
Alloue un nouveau tenseur qui est une copie d'un tableau donné de floats.
abstrait statique TBfloat16
vectorOf (valeurs flottantes...)
Alloue un nouveau tenseur pour stocker un vecteur de flotteurs.

Méthodes héritées

Méthodes publiques

résumé statique public TBfloat16 scalarOf (valeur flottante)

Alloue un nouveau tenseur pour stocker une seule valeur flottante.

Paramètres
valeur flotter pour stocker dans le nouveau tenseur
Retour
  • le nouveau tenseur

public static abstract TBfloat16 tensorOf ( Forme , Consommateur < TBfloat16 > dataInit)

Alloue un nouveau tenseur de la forme donnée et initialise ses données.

Paramètres
forme forme du tenseur à allouer
donnéesInit initialiseur de données tensorielles
Retour
  • le nouveau tenseur
Jetés
TensorFlowException si le tenseur ne peut pas être alloué ou initialisé

résumé statique public TBfloat16 tensorOf (forme de forme )

Alloue un nouveau tenseur de la forme donnée.

Paramètres
forme forme du tenseur à allouer
Retour
  • le nouveau tenseur

résumé statique public TBfloat16 tensorOf (forme de forme , données FloatDataBuffer )

Alloue un nouveau tenseur de la forme donnée, initialisé avec les données fournies.

Paramètres
forme forme du tenseur à allouer
données tampon de flotteurs pour initialiser le tenseur avec
Retour
  • le nouveau tenseur

résumé statique public TBfloat16 tensorOf ( NdArray <Float> src)

Alloue un nouveau tenseur qui est une copie d'un tableau donné de floats.

Le tenseur aura la même forme que le tableau source et ses données seront copiées.

Paramètres
src le tableau source donnant la forme et les données au nouveau tenseur
Retour
  • le nouveau tenseur

résumé statique public TBfloat16 vectorOf (valeurs flottantes...)

Alloue un nouveau tenseur pour stocker un vecteur de flotteurs.

Paramètres
valeurs flotte à stocker dans le nouveau tenseur
Retour
  • le nouveau tenseur