TBfloat16

interfaz pública TBfloat16

Cerebro tipo tensor flotante de 16 bits.

Este tipo se diferencia de TFloat16 en que trunca la mantisa de un flotante de 32 bits y conserva todos los bits del exponente para una conversión más rápida, mientras que este último reduce el exponente y tiene una mantisa más larga para mayor precisión.

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

Tenga en cuenta que algunas CPU admiten el formato bfloat16 de forma nativa, lo que puede dar como resultado un cálculo más rápido en comparación con TFloat16 cuando no se utilizan GPU.

Métodos públicos

TBfloat16 estático abstracto
escalarOf (valor flotante)
Asigna un nuevo tensor para almacenar un único valor flotante.
TBfloat16 estático abstracto
tensorOf ( forma , consumidor <TBfloat16> dataInit)
Asigna un nuevo tensor de la forma dada e inicializa sus datos.
TBfloat16 estático abstracto
tensorOf ( forma forma)
Asigna un nuevo tensor de la forma dada.
TBfloat16 estático abstracto
tensorOf (forma de forma , datos FloatDataBuffer )
Asigna un nuevo tensor de la forma dada, inicializado con los datos proporcionados.
TBfloat16 estático abstracto
tensorOf ( NdArray <Float> src)
Asigna un nuevo tensor que es una copia de una matriz determinada de flotantes.
TBfloat16 estático abstracto
vectorOf (flotante... valores)
Asigna un nuevo tensor para almacenar un vector de flotadores.

Métodos heredados

Métodos públicos

TBfloat16 abstracto estático público escalarOf (valor flotante)

Asigna un nuevo tensor para almacenar un único valor flotante.

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

tensorOf TBfloat16 abstracto estático público ( forma , consumidor <TBfloat16> dataInit)

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

Parámetros
forma forma del tensor para asignar
inicio de datos inicializador de datos tensoriales
Devoluciones
  • el nuevo tensor
Lanza
Excepción TensorFlow si el tensor no se puede asignar o inicializar

tensorOf TBfloat16 abstracto estático público ( forma )

Asigna un nuevo tensor de la forma dada.

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

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

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

Parámetros
forma forma del tensor para asignar
datos buffer de flotadores para inicializar el tensor con
Devoluciones
  • el nuevo tensor

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

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

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

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

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

Asigna un nuevo tensor para almacenar un vector de flotadores.

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