パブリック最終クラスSplitV
テンソルを 1 次元に沿って `num_split` テンソルに分割します。
パブリックメソッド
| static <T, U extends Number> SplitV <T> | |
| イテレータ<オペランド<T>> | イテレータ() |
| リスト<出力<T>> | 出力() `axis` に沿った部分を除き、形状が `value` の形状と一致するテンソル。サイズは `size_splits[i]` です。 |
継承されたメソッド
パブリックメソッド
public static SplitV <T> create (スコープscope、オペランド<T>値、オペランド<U>sizeSplits、オペランド<Integer>軸、Long numSplit)
新しい SplitV 操作をラップするクラスを作成するファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| 価値 | 分割するテンソル。 |
| サイズ分割 | 分割次元に沿った各出力テンソルのサイズを含むリスト。合計は、split_dim に沿った value の次元になる必要があります。次元が推定されることを示す -1 を 1 つ含めることができます。 |
| 軸 | 0-D。分割する次元。 `[-rank(value), Rank(value))` の範囲内である必要があります。 |
戻り値
- SplitV の新しいインスタンス