`indexes` に従って `params` 軸 `axis` からスライスを収集します。
`indices` は、任意の次元 (通常は 0-D または 1-D) の整数テンソルでなければなりません。形状 `params.shape[:axis] + indexs.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.Options | Gather のオプション属性 |
公開メソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
static Gather.Options | batchDims (長い batchDims) |
static <T, U extends Number, V extends Number> Gather <T> | |
出力<T> | 出力() `indexes` によって指定されたインデックスから収集された `params` からの値。 |
継承されたメソッド
公開メソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow 操作への入力は、別の TensorFlow 操作の出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static Gather <T> create (スコープスコープ、オペランド<T> パラメータ、オペランド<U> インデックス、オペランド<V> 軸、オプション...オプション)
新しい Gather 操作をラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
パラメータ | 値を収集するテンソル。ランク `axis + 1` 以上である必要があります。 |
インデックス | インデックス テンソル。 `[0, params.shape[axis])` の範囲内にある必要があります。 |
軸 | `index` を収集する `params` の軸。デフォルトは最初の次元です。負のインデックスをサポートします。 |
オプション | オプションの属性値を運ぶ |
戻り値
- Gather の新しいインスタンス
`indexes` に従って `params` 軸 `axis` からスライスを収集します。
`indices` は、任意の次元 (通常は 0-D または 1-D) の整数テンソルでなければなりません。形状 `params.shape[:axis] + indexs.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.Options | Gather のオプション属性 |
公開メソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
static Gather.Options | batchDims (長い batchDims) |
static <T, U extends Number, V extends Number> Gather <T> | |
出力<T> | 出力() `indexes` によって指定されたインデックスから収集された `params` からの値。 |
継承されたメソッド
公開メソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow 操作への入力は、別の TensorFlow 操作の出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static Gather <T> create (スコープスコープ、オペランド<T> パラメータ、オペランド<U> インデックス、オペランド<V> 軸、オプション...オプション)
新しい Gather 操作をラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
パラメータ | 値を収集するテンソル。ランク `axis + 1` 以上である必要があります。 |
インデックス | インデックス テンソル。 `[0, params.shape[axis])` の範囲内にある必要があります。 |
軸 | `index` を収集する `params` の軸。デフォルトは最初の次元です。負のインデックスをサポートします。 |
オプション | オプションの属性値を運ぶ |
戻り値
- Gather の新しいインスタンス