パブリックファイナルクラスExpandDims
テンソルの形状に1の次元を挿入します。
テンソル `input`が与えられると、この操作は` input`の形状の次元インデックス `axis`に次元1を挿入します。次元インデックス `axis`はゼロから始まります。 `axis`に負の数を指定すると、末尾から逆方向にカウントされます。
この操作は、バッチディメンションを単一の要素に追加する場合に役立ちます。たとえば、形状が `[height、width、channels]`の単一の画像がある場合、 `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の新しいインスタンス