TBfloat16

パブリックインターフェイスTBfloat16

Brain 16 ビット float テンソル型。

この型は、32 ビット浮動小数点数の仮数部を切り捨て、変換を高速化するためにすべての指数ビットを保持するのに対し、後者は指数を縮小し、精度を高めるために仮数部を長くするため、 TFloat16とは異なります。

Java には 16 ビットに適合する浮動小数点型がないため、JVM からこの型のテンソルに書き込まれるか読み取られる 32 ビット値ごとに変換 (精度が失われる可能性があります) が必要です。したがって、テンソルで多くの I/O 操作が予想される場合は、可能な限りTFloat32またはTFloat64データ型を使用することでパフォーマンスが向上します。

一部の CPU は bfloat16 形式をネイティブにサポートしているため、GPU が使用されていない場合はTFloat16と比較して計算が高速になる可能性があることに注意してください。

パブリックメソッド

抽象静的TBfloat16
scalarOf (浮動小数点値)
単一の float 値を格納するために新しいテンソルを割り当てます。
抽象静的TBfloat16
tensorOf ( Shapeシェイプ、Consumer< TBfloat16 > dataInit)
指定された形状の新しいテンソルを割り当て、そのデータを初期化します。
抽象静的TBfloat16
tensorOf ( Shapeシェイプ)
指定された形状の新しいテンソルを割り当てます。
抽象静的TBfloat16
tensorOf ( Shape形状、 FloatDataBufferデータ)
指定されたデータで初期化された、指定された形状の新しいテンソルを割り当てます。
抽象静的TBfloat16
tensorOf ( NdArray <Float> src)
指定された float 配列のコピーである新しいテンソルを割り当てます。
抽象静的TBfloat16
VectorOf (float... 値)
float のベクトルを格納するために新しいテンソルを割り当てます。

継承されたメソッド

パブリックメソッド

public static abstract TBfloat16 scalarOf (float 値)

単一の float 値を格納するために新しいテンソルを割り当てます。

パラメーター
価値新しいテンソルに格納する浮動小数点
戻り値
  • 新しいテンソル

public static abstract TBfloat16 tensorOf ( Shape形状、Consumer< TBfloat16 > dataInit)

指定された形状の新しいテンソルを割り当て、そのデータを初期化します。

パラメーター
割り当てるテンソルの形状
データ初期化テンソルデータ初期化子
戻り値
  • 新しいテンソル
投げる
TensorFlowExceptionテンソルを割り当てまたは初期化できない場合

public static abstract TBfloat16 tensorOf ( Shapeシェイプ)

指定された形状の新しいテンソルを割り当てます。

パラメーター
割り当てるテンソルの形状
戻り値
  • 新しいテンソル

public static abstract TBfloat16 tensorOf ( Shape形状、 FloatDataBufferデータ)

指定されたデータで初期化された、指定された形状の新しいテンソルを割り当てます。

パラメーター
割り当てるテンソルの形状
データテンソルを初期化するための浮動小数点数のバッファ
戻り値
  • 新しいテンソル

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

指定された float 配列のコピーである新しいテンソルを割り当てます。

テンソルはソース配列と同じ形状になり、そのデータがコピーされます。

パラメーター
送信元新しいテンソルに形状とデータを与えるソース配列
戻り値
  • 新しいテンソル

public static abstract TBfloat16 VectorOf (float... 値)

float のベクトルを格納するために新しいテンソルを割り当てます。

パラメーター
価値観新しいテンソルに格納する浮動小数点数
戻り値
  • 新しいテンソル