スプリットV

パブリック最終クラスSplitV

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

パブリックメソッド

static <T, U extends Number> SplitV <T>
create (スコープscope、オペランド<T>値、オペランド<U>sizeSplits、オペランド<Integer>軸、Long numSplit)
新しい SplitV 操作をラップするクラスを作成するファクトリ メソッド。
イテレータ<オペランド<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 の新しいインスタンス

public Iterator<オペランド<T>> iterator ()

public List<出力<T>>出力()

`axis` に沿った部分を除き、形状が `value` の形状と一致するテンソル。サイズは `size_splits[i]` です。