Gather

公開最終クラス集合

XLA Gather オペレーターをラップします。

https://www.tensorflow.org/xla/operation_semantics#gather

定数

OP_NAME TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリック ハンドルを返します。
static <T extends TType , U extends TNumber >収集<T>
create (スコープscope、オペランド<T>オペランド、オペランド<U>startIndices、オペランド<U>sliceSizes、文字列dimensionNumbers、Boolean indicesAreSorted)
新しい Gather オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<T>

継承されたメソッド

定数

パブリック静的最終文字列OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

定数値: "XlaGather"

パブリックメソッド

public Output <T> asOutput ()

テンソルのシンボリック ハンドルを返します。

TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。

public static Gather <T> create (スコープscope、オペランド<T>オペランド、オペランド<U>startIndices、オペランド<U>sliceSizes、文字列dimensionNumbers、Boolean indicesAreSorted)

新しい Gather オペレーションをラップするクラスを作成するためのファクトリ メソッド。

パラメーター
範囲現在のスコープ
オペランド収集元の配列。
startIndex収集したスライスの開始インデックスを含む配列。
スライスサイズlice_sizes[i] は、次元 i のスライスの境界です。
寸法番号シリアル化された xla::GatherDimensionNumbers プロト。
インデックスはソート済みインデックスがソートされているかどうかを示すブール値。
戻り値
  • Gather の新しいインスタンス

public出力<T>出力()