テンソルのシェイプに 1 の次元を挿入します。
テンソル `input` を指定すると、この操作は `input` の形状の次元インデックス `axis` に 1 の次元を挿入します。次元インデックス「axis」は 0 から始まります。 「axis」に負の数値を指定すると、最後から逆方向にカウントされます。
この操作は、単一の要素にバッチ ディメンションを追加する場合に便利です。たとえば、形状 `[height, width, Channels]` の画像が 1 つある場合、`expand_dims(image, 0)` を使用してそれを 1 つの画像のバッチにすると、形状 `[1, height] になります。 、幅、チャンネル]`。
他の例:
# 't' is a tensor of shape [2]
shape(expand_dims(t, 0)) ==> [1, 2]
shape(expand_dims(t, 1)) ==> [2, 1]
shape(expand_dims(t, -1)) ==> [2, 1]
# 't2' is a tensor of shape [2, 3, 5]
shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]
この操作には次のことが必要です。`-1-input.dims() <= dim <= input.dims()`
この操作は、サイズ 1 の次元を削除する `squeeze()` に関連しています。
パブリックメソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
static <T, U extends Number> ExpandDims <T> | |
出力<T> | 出力() 「input」と同じデータが含まれていますが、その形状にはサイズ 1 の追加次元が追加されています。 |
継承されたメソッド
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static ExpandDims <T> create (スコープスコープ、オペランド<T>入力、オペランド<U>軸)
新しい ExpandDims 操作をラップするクラスを作成するためのファクトリ メソッド。
パラメータ
範囲 | 現在のスコープ |
---|---|
軸 | 0-D (スカラー)。 `input` の形状を展開する次元インデックスを指定します。 `[-rank(input) - 1, Rank(input)]` の範囲内である必要があります。 |
返品
- ExpandDims の新しいインスタンス