TBfloat16

interfaccia pubblica TBfloat16

Tipo tensore float a 16 bit Brain.

Questo tipo differisce da TFloat16 in quanto tronca la mantissa di un float a 32 bit e preserva tutti i bit dell'esponente per una conversione più rapida, mentre quest'ultimo riduce l'esponente e ha una mantissa più lunga per una maggiore precisione.

Poiché in Java non esiste un tipo a virgola mobile che si adatti a 16 bit, è necessaria una conversione (con potenzialmente una perdita di precisione) per ogni valore a 32 bit scritto o letto su un tensore di questo tipo dalla JVM. Pertanto, se si prevedono molte operazioni di I/O su un tensore, le prestazioni verranno migliorate lavorando con i tipi di dati TFloat32 o TFloat64 quando possibile.

Tieni presente che alcune CPU supportano il formato bfloat16 in modo nativo, il che può comportare calcoli più rapidi rispetto a TFloat16 quando le GPU non vengono utilizzate.

Metodi pubblici

TBfloat16 statico astratto
scalarOf (valore float)
Alloca un nuovo tensore per memorizzare un singolo valore float.
TBfloat16 statico astratto
tensorOf ( Forma forma , Consumatore< TBfloat16 > dataInit)
Assegna un nuovo tensore della forma data e inizializza i suoi dati.
TBfloat16 statico astratto
tensorOf ( Forma forma)
Assegna un nuovo tensore della forma data.
TBfloat16 statico astratto
tensorOf (forma forma , dati FloatDataBuffer )
Assegna un nuovo tensore della forma data, inizializzato con i dati forniti.
TBfloat16 statico astratto
tensorOf ( NdArray <Float> src)
Assegna un nuovo tensore che è una copia di un dato array di float.
TBfloat16 statico astratto
vettoreOf (valori float...)
Assegna un nuovo tensore per memorizzare un vettore di float.

Metodi ereditati

Metodi pubblici

public static abstract TBfloat16 scalarOf (valore float)

Alloca un nuovo tensore per memorizzare un singolo valore float.

Parametri
valore float per memorizzare nel nuovo tensore
ritorna
  • il nuovo tensore

public static abstract TBfloat16 tensorOf ( Shape shape, Consumer< TBfloat16 > dataInit)

Assegna un nuovo tensore della forma data e inizializza i suoi dati.

Parametri
forma forma del tensore da allocare
datiInit inizializzatore di dati del tensore
ritorna
  • il nuovo tensore
Lancia
TensorFlowException se il tensore non può essere allocato o inizializzato

pubblico statico astratto TBfloat16 tensorOf ( Forma forma )

Assegna un nuovo tensore della forma data.

Parametri
forma forma del tensore da allocare
ritorna
  • il nuovo tensore

public static abstract TBfloat16 tensorOf ( Shape shape, dati FloatDataBuffer )

Assegna un nuovo tensore della forma data, inizializzato con i dati forniti.

Parametri
forma forma del tensore da allocare
dati buffer di float con cui inizializzare il tensore
ritorna
  • il nuovo tensore

pubblico statico astratto TBfloat16 tensorOf ( NdArray <Float> src)

Assegna un nuovo tensore che è una copia di un dato array di float.

Il tensore avrà la stessa forma dell'array di origine e i suoi dati verranno copiati.

Parametri
src l'array di origine che fornisce la forma e i dati al nuovo tensore
ritorna
  • il nuovo tensore

public static abstract TBfloat16 vectorOf (valori float...)

Assegna un nuovo tensore per memorizzare un vettore di float.

Parametri
valori galleggia per essere memorizzato nel nuovo tensore
ritorna
  • il nuovo tensore