Gather

公開最終クラス集合

`params` 軸 `axis` から `indices` に従ってスライスを収集します。

`indices` は、任意の次元 (通常は 0 次元または 1 次元) の整数テンソルでなければなりません。形状 `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]` を持つ出力テンソルを生成します。ここで、 

# Scalar indices (output is rank(params) - 1).
     output[a_0, ..., a_n, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices, b_0, ..., b_n]
 
     # Vector indices (output is rank(params)).
     output[a_0, ..., a_n, i, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
 
     # Higher rank indices (output is rank(params) + rank(indices) - 1).
     output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
       params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]

CPU では、範囲外のインデックスが見つかった場合、エラーが返されることに注意してください。 GPU では、範囲外のインデックスが見つかった場合、対応する出力値に 0 が格納されます。

`tf.batch_gather` およびtf.gather_ndも参照してください。

ネストされたクラス

クラス収集オプションGatherのオプション属性

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリック ハンドルを返します。
静的Gather.Options
バッチディム(長いバッチディム)
static <T, U は数値を拡張、V は数値を拡張>収集<T>
create (スコープscope、オペランド<T>パラメータ、オペランド<U>インデックス、オペランド<V>軸、オプション...オプション)
新しい Gather オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<T>
出力()
`params` の値は、`params.shape[:axis] + indices.shape + params.shape[axis + 1:]` の形状を持つ、`indices` で指定されたインデックスから収集されます。

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

public static Gather.OptionsバッチDims (LongバッチDims)

public static Gather <T> create (スコープスコープ、オペランド<T> パラメータ、オペランド<U> インデックス、オペランド<V> 軸、オプション...オプション)

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

パラメータ
範囲現在のスコープ
パラメータ値を収集するテンソル。少なくともランク「axis + 1」でなければなりません。
インデックスインデックステンソル。範囲 `[0, params.shape[axis])` 内にある必要があります。
軸「インデックス」を収集するための「params」内の軸。デフォルトは最初の次元です。負のインデックスをサポートします。
オプションオプションの属性値を持ちます
返品
  • Gather の新しいインスタンス

public出力<T>出力()

`params` の値は、`params.shape[:axis] + indices.shape + params.shape[axis + 1:]` の形状を持つ、`indices` で指定されたインデックスから収集されます。