TFloat16

interfaz pública TFloat16

Tipo de tensor flotante de 16 bits de precisión media IEEE-754.

Dado que no existe un tipo de punto flotante que quepa en 16 bits en Java, se requiere una conversión (con una pérdida potencial de precisión) para cada valor de 32 bits escrito o leído en un tensor de este tipo de la JVM. Por lo tanto, si se esperan muchas operaciones de E / S en un tensor, el rendimiento mejorará si se trabaja con los tipos de datos TFloat32 o TFloat64 siempre que sea posible.

Además, los tensores TFloat16 normalmente funcionan mejor si están ubicados en la memoria de la GPU, ya que la mayoría de las CPU no admiten este formato de forma nativa. Para el cálculo de la CPU en flotantes de 16 bits, el tipo de tensor TBfloat16 podría ser una mejor opción.

Métodos públicos

TFloat16 estático abstracto
scalarOf (valor flotante)
Asigna un nuevo tensor para almacenar un solo valor flotante.
TFloat16 estático abstracto
tensorOf ( forma de forma)
Asigna un nuevo tensor de la forma dada.
TFloat16 estático abstracto
tensorOf (forma Shape , datos FloatDataBuffer )
Asigna un nuevo tensor de la forma dada, inicializado con los datos proporcionados.
TFloat16 estático abstracto
tensorOf ( Shape shape, Consumer < TFloat16 > dataInit)
Asigna un nuevo tensor de la forma dada e inicializa sus datos.
TFloat16 estático abstracto
tensorOf ( NdArray <Float> src)
Asigna un nuevo tensor que es una copia de una matriz dada de flotantes.
TFloat16 estático abstracto
vectorOf (valores flotantes ...)
Asigna un nuevo tensor para almacenar un vector de flotantes.

Métodos heredados

Métodos públicos

TFloat16 scalarOf abstracto estático público (valor flotante)

Asigna un nuevo tensor para almacenar un solo valor flotante.

Parámetros
valor flotar para almacenar en el nuevo tensor
Devoluciones
  • el nuevo tensor

público estático abstracto TFloat16 tensorOf ( forma de forma)

Asigna un nuevo tensor de la forma dada.

Parámetros
forma forma del tensor a asignar
Devoluciones
  • el nuevo tensor

público estático abstracto TFloat16 tensorOf ( forma de forma, datos FloatDataBuffer )

Asigna un nuevo tensor de la forma dada, inicializado con los datos proporcionados.

Parámetros
forma forma del tensor a asignar
datos búfer de flotadores para inicializar el tensor con
Devoluciones
  • el nuevo tensor

público estático abstracto TFloat16 tensorOf ( Shape shape, Consumer < TFloat16 > dataInit)

Asigna un nuevo tensor de la forma dada e inicializa sus datos.

Parámetros
forma forma del tensor a asignar
dataInit inicializador de datos de tensor
Devoluciones
  • el nuevo tensor
Lanza
TensorFlowException si el tensor no se puede asignar o inicializar

público estático abstracto TFloat16 tensorOf ( NdArray <Float> src)

Asigna un nuevo tensor que es una copia de una matriz dada de flotantes.

El tensor tendrá la misma forma que la matriz de origen y se copiarán sus datos.

Parámetros
src la matriz de origen que da la forma y los datos al nuevo tensor
Devoluciones
  • el nuevo tensor

vectorOf abstracto estático público TFloat16 (valores ... flotantes)

Asigna un nuevo tensor para almacenar un vector de flotantes.

Parámetros
valores flota para almacenar en el nuevo tensor
Devoluciones
  • el nuevo tensor