TFloat16

interfaccia pubblica TFloat16

Tipo di tensore float a 16 bit a mezza precisione IEEE-754.

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.

Inoltre, i tensori TFloat16 normalmente funzionano meglio se si trovano nella memoria della GPU poiché la maggior parte delle CPU non supporta questo formato in modo nativo. Per il calcolo della CPU su float a 16 bit, il tipo di tensore TBfloat16 potrebbe essere un'opzione migliore.

Metodi pubblici

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

Metodi ereditati

Metodi pubblici

public static abstract TFloat16 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

pubblico statico astratto TFloat16 tensorOf ( Forma forma )

Assegna un nuovo tensore della forma data.

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

public static abstract TFloat16 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

public static abstract TFloat16 tensorOf ( Shape shape, Consumer< TFloat16 > 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 TFloat16 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

pubblico statico astratto TFloat16 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