テンソルフロー::作戦::集める
#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 )