tensorflow:: ops:: SparseSplit

#include <sparse_ops.h>

SparseTensor 1次元に沿ってnum_splitテンソルに分割します。

まとめ

shape[split_dim]がnum_splitの整数倍でない場合、スライス[0 : shape[split_dim] % num_split]には 1 つの次元が追加されます。たとえば、 split_dim = 1 、 num_split = 2で入力が

input_tensor = shape = [2, 7]
[    a   d e  ]
[b c          ]

出力テンソルをグラフで示すと次のようになります。

output_tensor[0] = shape = [2, 4]
[    a  ]
[b c    ]

output_tensor[1] = shape = [2, 3]
[ d e  ]
[      ]

議論:

  • スコープ:スコープオブジェクト
  • split_dim: 0-D。分割する次元。範囲は[0, rank(shape))である必要があります。
  • インデックス: 2次元テンソルは、疎テンソルのインデックスを表します。
  • values: 1次元テンソルは、疎テンソルの値を表します。
  • shape: 1次元テンソルは、疎テンソルの形状を表します。 output indices: 1次元テンソルのリストは、出力疎テンソルのインデックスを表します。
  • num_split: 分割方法の数。

返品:

  • OutputList出力インデックス
  • OutputList output_values: 出力スパーステンソルの値を表す1次元テンソルのリスト。
  • OutputList output_shape: 1次元テンソルのリストは、出力スパーステンソルの形状を表します。

コンストラクターとデストラクタ

SparseSplit (const :: tensorflow::Scope & scope, :: tensorflow::Input split_dim, :: tensorflow::Input indices, :: tensorflow::Input values, :: tensorflow::Input shape, int64 num_split)

公開属性

operation
output_indices
output_shape
output_values

公開属性

手術

Operation operation

出力インデックス

::tensorflow::OutputList output_indices

出力形状

::tensorflow::OutputList output_shape

出力値

::tensorflow::OutputList output_values

公的機能

スパース分割

 SparseSplit(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input split_dim,
  ::tensorflow::Input indices,
  ::tensorflow::Input values,
  ::tensorflow::Input shape,
  int64 num_split
)