TPUPartitionedOutputV2

パブリック最終クラスTPUPartitionedOutputV2

XLA によって分割されるテンソルを分割されたリストに逆多重化する演算

XLA 計算の外で出力します。 NDシャーディングをサポートします。

パブリックメソッド

静的 <T> TPUPartitionedOutputV2 <T>
create (スコープscope、オペランド<T>入力、Long numSplits、List<Long>partitionDims)
新しい TPUPartitionedOutputV2 オペレーションをラップするクラスを作成するためのファクトリ メソッド。
イテレータ<オペランド<T>>
リスト<出力<T>>
出力()
同じ形状を持つ分割された出力のリスト。

継承されたメソッド

パブリックメソッド

public static TPUPartitionedOutputV2 <T> create (スコープscope、オペランド<T>入力、Long numSplits、List<Long>partitionDims)

新しい TPUPartitionedOutputV2 オペレーションをラップするクラスを作成するためのファクトリ メソッド。

パラメーター
範囲現在のスコープ
入力分割されたテンソルの完全な形状を表すテンソル。
パーティションディム各ディメンションがどのように分割されるかを説明する整数のリスト。空の場合は、入力が複製されていることを示します。
戻り値
  • TPUPartitionedOutputV2 の新しいインスタンス

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

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

同じ形状を持つ分割された出力のリスト。