テンソルフロー::作戦::スパーススプリット

#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  ]
[      ]

引数:

  • スコープ:スコープオブジェクト
  • 分割寸法: 0-D。分割する次元。 [0, rank(shape))の範囲内である必要があります。
  • インデックス: 2 次元テンソルは、スパース テンソルのインデックスを表します。
  • 値: 1 次元テンソルはスパース テンソルの値を表します。
  • 形状: 1-D。 tensor はスパース テンソルの形状を表します。出力インデックス: 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
)