パブリック最終クラスReadVariableXlaSplitND
リソース変数の入力テンソルをすべての次元にわたって分割します。
指定された num_splits 属性に基づいてリソース変数入力テンソルを分割し、オプションでスライスをパディングし、スライスを返す演算です。スライスは行優先の順序で返されます。
この演算は TPU ブリッジ経由で生成される場合があります。
たとえば、「input」テンソルの場合:
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
[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> | |
イテレータ<オペランド<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 static ReadVariableXlaSplitND.Optionsパディング(List<Long> パディング)
パラメータ
パディング | 分割前に適用する入力テンソルの次元ごとの右パディングのオプションのリスト。これを使用して、次元を均等に割り切れるようにすることができます。 |
---|