AssignVariableXlaConcatND

パブリック最終クラスAssignVariableXlaConcatND

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

マージする操作は、指定された num_splits 属性に基づいて入力テンソルをスライスし、必要に応じてパディングを取り除き、マージされたテンソルをパディングなしでリソース変数に書き込みます。

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

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

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

ネストされたクラス

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

パブリックメソッド

静的 <T> AssignVariableXlaConcatND
create (スコープスコープ、オペランド<?> リソース、Iterable<オペランド<T>> 入力、List<Long> numConcats、オプション...オプション)
新しい AssignVariableXlaConcatND 操作をラップするクラスを作成するファクトリ メソッド。
静的AssignVariableXlaConcatND.Options
パディング(List<Long> パディング)

継承されたメソッド

パブリックメソッド

public static AssignVariableXlaConcatND create (スコープスコープ、オペランド<?> リソース、Iterable<オペランド<T>> 入力、List<Long> numConcats、オプション...オプション)

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

パラメータ
範囲現在のスコープ
リソースすべての次元にわたって連結された入力テンソルのリソース変数。 } in_arg { 名前: "入力" 説明: <
numConcatsディメンションごとのマージ方法の数。
オプションオプションの属性値を持ちます
返品
  • AssignVariableXlaConcatND の新しいインスタンス

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

パラメータ
パディング最終的にマージされたテンソルから除去する次元ごとの右パディングのオプションのリスト。これらのパディングは、パディングを除去する前にマージされた結果の次元サイズを超えてはなりません。