TBfloat16

общедоступный интерфейс TBfloat16

Мозг 16-битного плавающего тензорного типа.

Этот тип отличается от TFloat16 тем, что он усекает мантиссу 32-битного числа с плавающей запятой и сохраняет все биты экспоненты для более быстрого преобразования, тогда как последний сжимает экспоненту и имеет более длинную мантиссу для большей точности.

Поскольку в Java не существует типа с плавающей запятой, который умещался бы в 16 бит, преобразование (потенциально с потерей точности) требуется для каждого 32-битного значения, записанного или прочитанного в тензоре этого типа из JVM. Следовательно, если на тензоре ожидается много операций ввода-вывода, производительность будет повышена за счет работы с типами данных TFloat32 или TFloat64 , когда это возможно.

Обратите внимание, что некоторые процессоры изначально поддерживают формат bfloat16, что может привести к более быстрым вычислениям по сравнению с TFloat16 , когда графические процессоры не используются.

Публичные методы

абстрактный статический TBfloat16
скалярOf (число с плавающей запятой)
Выделяет новый тензор для хранения одного значения с плавающей запятой.
абстрактный статический TBfloat16
tensorOf (Форма формы , Consumer < TBfloat16 > dataInit)
Выделяет новый тензор заданной формы и инициализирует его данные.
абстрактный статический TBfloat16
tensorOf (форма фигуры )
Выделяет новый тензор заданной формы.
абстрактный статический TBfloat16
tensorOf (форма фигуры , данные FloatDataBuffer )
Выделяет новый тензор заданной формы, инициализируемый предоставленными данными.
абстрактный статический TBfloat16
tensorOf ( NdArray <Float> источник)
Выделяет новый тензор, который является копией заданного массива чисел с плавающей запятой.
абстрактный статический TBfloat16
VectorOf (значения с плавающей запятой)
Выделяет новый тензор для хранения вектора чисел с плавающей запятой.

Унаследованные методы

Публичные методы

общедоступный статический абстрактный TBfloat16 скалярOf (значение с плавающей запятой)

Выделяет новый тензор для хранения одного значения с плавающей запятой.

Параметры
ценить float для хранения в новом тензоре
Возврат
  • новый тензор

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

Выделяет новый тензор заданной формы и инициализирует его данные.

Параметры
форма форма тензора для распределения
данныеинит инициализатор тензорных данных
Возврат
  • новый тензор
Броски
TensorFlowException если тензор не может быть выделен или инициализирован

общедоступная статическая абстрактная TBfloat16 tensorOf (форма формы )

Выделяет новый тензор заданной формы.

Параметры
форма форма тензора для распределения
Возврат
  • новый тензор

public static Abstract TBfloat16 tensorOf (форма фигуры , данные FloatDataBuffer )

Выделяет новый тензор заданной формы, инициализируемый предоставленными данными.

Параметры
форма форма тензора для распределения
данные буфер поплавков для инициализации тензора с помощью
Возврат
  • новый тензор

общедоступный статический абстрактный TBfloat16 tensorOf ( NdArray <Float> src)

Выделяет новый тензор, который является копией заданного массива чисел с плавающей запятой.

Тензор будет иметь ту же форму, что и исходный массив, и его данные будут скопированы.

Параметры
источник исходный массив, передающий форму и данные новому тензору
Возврат
  • новый тензор

общедоступный статический абстрактный TBfloat16 VectorOf (значения с плавающей запятой...)

Выделяет новый тензор для хранения вектора чисел с плавающей запятой.

Параметры
ценности плавает для хранения в новом тензоре
Возврат
  • новый тензор