TBfloat16

interfejs publiczny TBfloat16

Mózg 16-bitowy typ tensora zmiennoprzecinkowego.

Ten typ różni się od TFloat16 tym, że obcina mantysę 32-bitowego float i zachowuje wszystkie bity wykładnika w celu szybszej konwersji, podczas gdy ten drugi zmniejsza wykładnik i ma dłuższą mantysę dla większej precyzji.

Ponieważ w Javie nie ma typu zmiennoprzecinkowego mieszczącego się w 16 bitach, wymagana jest konwersja (z potencjalną utratą precyzji) dla każdej 32-bitowej wartości zapisanej lub odczytanej na tensorze tego typu z JVM. Dlatego też, jeśli na tensorze można spodziewać się wielu operacji we/wy, wydajność zostanie poprawiona poprzez pracę z typami danych TFloat32 lub TFloat64 , jeśli to możliwe.

Należy pamiętać, że niektóre procesory natywnie obsługują format bfloat16, co może skutkować szybszymi obliczeniami w porównaniu do TFloat16 , gdy nie są używane procesory graficzne.

Metody publiczne

abstrakcyjny statyczny TBfloat16
scalarOf (wartość zmiennoprzecinkowa)
Przydziela nowy tensor do przechowywania pojedynczej wartości zmiennoprzecinkowej.
abstrakcyjny statyczny TBfloat16
tensorOf ( Kształt kształtu, Konsument < TBfloat16 > dataInit)
Przydziela nowy tensor danego kształtu i inicjuje jego dane.
abstrakcyjny statyczny TBfloat16
tensorOf (kształt kształtu )
Przydziela nowy tensor o zadanym kształcie.
abstrakcyjny statyczny TBfloat16
tensorOf (kształt kształtu , dane FloatDataBuffer )
Przydziela nowy tensor o zadanym kształcie, inicjowany dostarczonymi danymi.
abstrakcyjny statyczny TBfloat16
tensorOf ( NdArray <Float> src)
Przydziela nowy tensor, który jest kopią danej tablicy zmiennych.
abstrakcyjny statyczny TBfloat16
wektorOf (wartości zmiennoprzecinkowe)
Przydziela nowy tensor do przechowywania wektora liczb zmiennoprzecinkowych.

Metody dziedziczone

Metody publiczne

publiczne statyczne streszczenie TBfloat16 scalarOf (wartość zmiennoprzecinkowa)

Przydziela nowy tensor do przechowywania pojedynczej wartości zmiennoprzecinkowej.

Parametry
wartość float do przechowywania w nowym tensorze
Zwroty
  • nowy tensor

publiczna statyczna abstrakcja TBfloat16 tensorOf ( Kształt kształtu , Konsument< TBfloat16 > dataInit)

Przydziela nowy tensor danego kształtu i inicjuje jego dane.

Parametry
kształt kształt tensora do przydzielenia
dataInit inicjator danych tensora
Zwroty
  • nowy tensor
Rzuca
Wyjątek TensorFlow jeśli nie można przydzielić lub zainicjować tensora

publiczna statyczna abstrakcja TBfloat16 tensorOf (kształt kształtu )

Przydziela nowy tensor o zadanym kształcie.

Parametry
kształt kształt tensora do przydzielenia
Zwroty
  • nowy tensor

publiczna statyczna abstrakcja TBfloat16 tensorOf (kształt kształtu , dane FloatDataBuffer )

Przydziela nowy tensor o zadanym kształcie, inicjowany dostarczonymi danymi.

Parametry
kształt kształt tensora do przydzielenia
dane bufor elementów zmiennoprzecinkowych, za pomocą którego można zainicjować tensor
Zwroty
  • nowy tensor

publiczna statyczna abstrakcja TBfloat16 tensorOf ( NdArray <Float> src)

Przydziela nowy tensor, który jest kopią danej tablicy zmiennych.

Tensor będzie miał taki sam kształt jak tablica źródłowa, a jego dane zostaną skopiowane.

Parametry
źródło tablica źródłowa nadająca kształt i dane nowemu tensorowi
Zwroty
  • nowy tensor

publiczna statyczna abstrakcja TBfloat16 wektorOf (wartości zmiennoprzecinkowe)

Przydziela nowy tensor do przechowywania wektora liczb zmiennoprzecinkowych.

Parametry
wartości float do przechowywania w nowym tensorze
Zwroty
  • nowy tensor