テンソルフロー::作戦::集める
#include <array_ops.h> indicesに従ってparamsからスライスを収集します。
まとめ
indices 、任意の次元 (通常は 0 次元または 1 次元) の整数テンソルでなければなりません。形状indices.shape + params.shape[1:]を使用して出力テンソルを生成します。ここで、
# Scalar indices
output[:, ..., :] = params[indices, :, ... :]# Vector indices output[i, :, ..., :] = params[indices[i], :, ... :]
# Higher rank indices
output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :] indicesが並べ替えであり、 len(indices) == params.shape[0]の場合、この操作はそれに応じてparamsを並べ替えます。
validate_indices : 非推奨。この操作が CPU に割り当てられている場合、 indicesの値は常に範囲内であることが検証されます。 GPU に割り当てられた場合、範囲外のインデックスは安全ではあるものの、エラーの発生を含む不特定の動作を引き起こします。

引数:
- スコープ:スコープオブジェクト
戻り値:
-
Output: 出力テンソル。
コンストラクターとデストラクター | |
|---|---|
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices) | |
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, const Gather::Attrs & attrs) |
パブリック属性 | |
|---|---|
operation | |
output | |
公共機能 | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
パブリック静的関数 | |
|---|---|
ValidateIndices (bool x) | |
構造体 | |
|---|---|
| tensorflow:: ops:: Gather:: Attrs | Gatherのオプションの属性セッター。 |
パブリック属性
手術
Operation operation
出力
::tensorflow::Output output
公共機能
集める
Gather( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices )
集める
Gather( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices, const Gather::Attrs & attrs )
ノード
::tensorflow::Node * node() const
演算子::tensorflow::入力
operator::tensorflow::Input() const
演算子::tensorflow::出力
operator::tensorflow::Output() const
パブリック静的関数
インデックスの検証
Attrs ValidateIndices( bool x )