Menyisipkan dimensi 1 ke dalam bentuk tensor.
Diberikan `input` tensor, operasi ini menyisipkan dimensi 1 pada indeks dimensi `sumbu` dari bentuk `input`. Indeks dimensi `sumbu` dimulai dari nol; jika Anda menentukan angka negatif untuk `sumbu`, angka tersebut dihitung mundur dari akhir.
Operasi ini berguna jika Anda ingin menambahkan dimensi batch ke satu elemen. Misalnya, jika Anda memiliki satu gambar berbentuk `[tinggi, lebar, saluran]`, Anda dapat menjadikannya kumpulan 1 gambar dengan `expand_dims(gambar, 0)`, yang akan menghasilkan bentuk `[1, tinggi , lebar, saluran]`.
Contoh lain:
# '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() <= redup <= masukan.dims()`
Operasi ini terkait dengan `squeeze()`, yang menghapus dimensi ukuran 1.
Konstanta
Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
statis <T memperluas TType > ExpandDims <T> | |
Keluaran <T> | keluaran () Berisi data yang sama dengan `input`, namun bentuknya memiliki dimensi tambahan sebesar 1 yang ditambahkan. |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Metode Publik
Keluaran publik <T> sebagai Keluaran ()
Mengembalikan pegangan simbolis tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static ExpandDims <T> buat ( Lingkup lingkup , masukan Operan <T>, Operan <? extends TNumber > sumbu)
Metode pabrik untuk membuat kelas yang membungkus operasi ExpandDims baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
sumbu | 0-D (skalar). Menentukan indeks dimensi untuk memperluas bentuk `input`. Harus berada dalam rentang `[-peringkat(input) - 1, peringkat(input)]`. |
Kembali
- contoh baru dari ExpandDims
Keluaran publik <T> keluaran ()
Berisi data yang sama dengan `input`, namun bentuknya memiliki dimensi tambahan sebesar 1 yang ditambahkan.