テンソルフロー::作戦::集める

#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
)