XlaSplitND

パブリック最終クラスXlaSplitND

入力テンソルをすべての次元にわたって分割します。

指定された num_splits 属性に基づいて入力テンソルをスライスし、オプションでスライスをパディングし、スライスを返す演算です。スライスは行優先の順序で返されます。

この演算は TPU ブリッジ経由で生成される場合があります。

たとえば、「input」テンソルの場合: 

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
`num_splits`: 
[2, 2]
と「パディング」: 
[1, 1]
期待される「出力」は次のとおりです。 
[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]

ネストされたクラス

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

パブリックメソッド

静的 <T> XlaSplitND <T>
create (スコープscope、オペランド<T>入力、Long N、List<Long> numSplits、オプション...オプション)
新しい XlaSplitND 操作をラップするクラスを作成するファクトリ メソッド。
イテレータ<オペランド<T>>
イテレータ()
リスト<出力<T>>
出力()
静的XlaSplitND.Options
パディング(List<Long> パディング)

継承されたメソッド

パブリックメソッド

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

新しい XlaSplitND 操作をラップするクラスを作成するファクトリ メソッド。

パラメータ
範囲現在のスコープ
入力すべての次元にわたって分割する入力テンソル。 } out_arg { 名前: "出力" 説明: <
分割数次元ごとの分割方法の数。形状の寸法は均等に分割できる必要があります。
オプションオプションの属性値を持ちます
返品
  • XlaSplitND の新しいインスタンス

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

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

public static XlaSplitND.Optionsパディング(List<Long> パディング)

パラメータ
パディング分割前に適用する入力テンソルの次元ごとの右パディングのオプションのリスト。これを使用して、次元を均等に割り切れるようにすることができます。