XlaConcatND

パブリック最終クラスXlaConcatND

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

マージする操作は、指定された 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]]

ネストされたクラス

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

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリック ハンドルを返します。
静的 <T> XlaConcatND <T>
create ( Scopeスコープ、Iterable< Operand <T>> inputs、List<Long> numConcats、 Options...オプション)
新しい XlaConcatND 操作をラップするクラスを作成するファクトリ メソッド。
出力<T>
出力()
静的XlaConcatND.Options
パディング(List<Long> パディング)

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

テンソルのシンボリック ハンドルを返します。

TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。

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

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

パラメータ
範囲現在のスコープ
入力行優先の順序でテンソル スライスを入力し、すべての次元にわたってマージします。すべての入力は同じ形状でなければなりません。 } out_arg { 名前: "出力" 説明: <
numConcatsディメンションごとのマージ方法の数。
オプションオプションの属性値を持ちます
返品
  • XlaConcatND の新しいインスタンス

public出力<T>出力()

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

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