TBfloat16

interface pública TBfloat16

Tipo tensor flutuante de 16 bits do cérebro.

Este tipo difere do TFloat16 , pois trunca a mantissa de um float de 32 bits e preserva todos os bits do expoente para uma conversão mais rápida, enquanto o último encolhe o expoente e tem uma mantissa mais longa para maior precisão.

Como não há tipo de ponto flutuante que caiba em 16 bits em Java, uma conversão (com potencialmente uma perda de precisão) é necessária para cada valor de 32 bits escrito ou lido em um tensor desse tipo a partir da JVM. Portanto, se um monte de I / O operações estão a ser esperado em um tensor, performances será melhorada através do trabalho com TFloat32 ou TFloat64 tipos de dados sempre que possível.

Observe que algumas CPUs suportam o formato bfloat16 nativamente, o que pode resultar em um cálculo mais rápido em comparação com TFloat16 quando as GPUs não são usadas.

Métodos Públicos

TBfloat estático abstrato 16
scalarOf (valor flutuante)
Aloca um novo tensor para armazenar um único valor flutuante.
TBfloat estático abstrato 16
tensorOf ( Shape shape, Consumer < TBfloat16 > dataInit)
Aloca um novo tensor da forma fornecida e inicializa seus dados.
TBfloat estático abstrato 16
tensorOf ( forma da forma)
Aloca um novo tensor da forma fornecida.
TBfloat estático abstrato 16
tensorOf ( Shape shape, FloatDataBuffer data)
Aloca um novo tensor da forma fornecida, inicializado com os dados fornecidos.
TBfloat estático abstrato 16
tensorOf ( NdArray <Float> src)
Aloca um novo tensor que é uma cópia de um dado array de floats.
TBfloat estático abstrato 16
vectorOf (float ... valores)
Aloca um novo tensor para armazenar um vetor de flutuadores.

Métodos herdados

Métodos Públicos

public static abstract TBfloat16 scalarOf (valor flutuante)

Aloca um novo tensor para armazenar um único valor flutuante.

Parâmetros
valor flutuar para armazenar no novo tensor
Devoluções
  • o novo tensor

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

Aloca um novo tensor da forma fornecida e inicializa seus dados.

Parâmetros
forma forma do tensor para alocar
dataInit inicializador de dados de tensor
Devoluções
  • o novo tensor
Lança
TensorFlowException se o tensor não pode ser alocado ou inicializado

public static abstract TBfloat16 tensorOf ( Shape shape)

Aloca um novo tensor da forma fornecida.

Parâmetros
forma forma do tensor para alocar
Devoluções
  • o novo tensor

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

Aloca um novo tensor da forma fornecida, inicializado com os dados fornecidos.

Parâmetros
forma forma do tensor para alocar
dados buffer de flutuadores para inicializar o tensor com
Devoluções
  • o novo tensor

public static abstract TBfloat16 tensorOf ( NdArray <Float> src)

Aloca um novo tensor que é uma cópia de um dado array de floats.

O tensor terá a mesma forma da matriz de origem e seus dados serão copiados.

Parâmetros
src a matriz de origem dando a forma e os dados para o novo tensor
Devoluções
  • o novo tensor

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

Aloca um novo tensor para armazenar um vetor de flutuadores.

Parâmetros
valores flutua para armazenar no novo tensor
Devoluções
  • o novo tensor