パブリック最終クラスTPUPartitionedOutputV2
XLA によって分割されるテンソルを分割されたリストに逆多重化する演算
XLA 計算の外で出力します。 NDシャーディングをサポートします。
パブリックメソッド
静的 <T> TPUPartitionedOutputV2 <T> | |
イテレータ<オペランド<T>> | イテレータ() |
リスト<出力<T>> | 出力() 同じ形状を持つ分割された出力のリスト。 |
継承されたメソッド
パブリックメソッド
public static TPUPartitionedOutputV2 <T> create (スコープscope、オペランド<T>入力、Long numSplits、List<Long>partitionDims)
新しい TPUPartitionedOutputV2 オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
入力 | 分割されたテンソルの完全な形状を表すテンソル。 |
パーティションディム | 各ディメンションがどのように分割されるかを説明する整数のリスト。空の場合は、入力が複製されていることを示します。 |
戻り値
- TPUPartitionedOutputV2 の新しいインスタンス