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]
および `paddings`: 
[1, 1]
の場合、期待される `outputs` は次のようになります: 
[[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> パディング)

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