TPUパーティション化された出力

パブリック最終クラスTPUPartitionedOutput

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

XLA 計算の外で出力します。

ネストされたクラス

クラスTPUPartitionedOutput.Options TPUPartitionedOutputのオプションの属性

パブリックメソッド

静的 <T> TPUPartitionedOutput <T>
create (スコープスコープ、オペランド<T> 入力、Long numSplits、オプション...オプション)
新しい TPUPartitionedOutput オペレーションをラップするクラスを作成するためのファクトリ メソッド。
イテレータ<オペランド<T>>
イテレータ()
リスト<出力<T>>
出力()
同じ形状を持つ必要がある分割された入力のリスト。
静的TPUPartitionedOutput.Options
パーティションDim (長いパーティションDim)

継承されたメソッド

パブリックメソッド

public static TPUPartitionedOutput <T> create (スコープスコープ、オペランド<T>入力、Long numSplits、オプション...オプション)

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

パラメーター
範囲現在のスコープ
入力分割されたテンソルの完全な形状を表すテンソル。
オプションオプションの属性値を持ちます
戻り値
  • TPUPartitionedOutput の新しいインスタンス

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

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

同じ形状を持つ必要がある分割された入力のリスト。

public static TPUPartitionedOutput.OptionspartitionDim ( LongpartitionDim)

パラメーター
パーティションディム整数はどの次元が分割されるかを表します。