TBfloat16

interfaz pública TBfloat16

Tipo de tensor de flotación cerebral 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 el último reduce el exponente y tiene una mantisa más larga para una mayor precisión.

Dado que no existe un tipo de punto flotante que quepa en 16 bits en Java, se requiere una conversión (con una posible pérdida 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á al trabajar con 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 resultar en 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
scalarOf (valor flotante)
Asigna un nuevo tensor para almacenar un solo valor flotante.
TBfloat16 estático abstracto
tensorOf ( Shape shape, Consumer < TBfloat16 > dataInit)
Asigna un nuevo tensor de la forma dada e inicializa sus datos.
TBfloat16 estático abstracto
tensorOf ( forma de 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 dada de flotantes.
TBfloat16 estático abstracto
vectorOf (valores flotantes ...)
Asigna un nuevo tensor para almacenar un vector de flotantes.

Métodos heredados

Métodos públicos

resumen estático público TBfloat16 scalarOf (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

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

public static abstract TBfloat16 tensorOf ( Shape shape)

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 TBfloat16 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 TBfloat16 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

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

Asigna un nuevo tensor para almacenar un vector de flotantes.

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