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

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 次元テンソルのリストは、出力スパース テンソルの形状を表します。

パブリック属性

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
)