ReadVariableXlaSplitND

パブリック最終クラスReadVariableXlaSplitND

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

指定された 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]]
 

ネストされたクラス

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

パブリックメソッド

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

継承されたメソッド

パブリックメソッド

public static ReadVariableXlaSplitND <T> create (スコープスコープ、オペランド<?> リソース、Class<T> T、Long N、List<Long> numSplits、 Options...オプション)

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

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

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

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

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

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