PartitionedOutput

パブリック最終クラスPartitionedOutput

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

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

ネストされたクラス

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

定数

弦OP_NAME TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

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

継承されたメソッド

定数

パブリック静的最終文字列OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

定数値: "TPUPartitionedOutput"

パブリックメソッド

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

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

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

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

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

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

public static PartitionedOutput.OptionspartitionDim ( LongpartitionDim)

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