aliran tensor:: operasi:: MaxPoolDenganArgmax
#include <nn_ops.h>
Melakukan pengumpulan maksimal pada input dan output baik nilai maupun indeks maksimal.
Ringkasan
Indeks di argmax
diratakan, sehingga nilai maksimum pada posisi [b, y, x, c]
menjadi indeks rata: (y * width + x) * channels + c
jika include_batch_in_index
adalah False; ((b * height + y) * width + x) * channels + c
jika include_batch_in_index
Benar.
Indeks yang dikembalikan selalu dalam [0, height) x [0, width)
sebelum diratakan, meskipun ada bantalan dan jawaban yang benar secara matematis berada di luar (negatif atau terlalu besar). Ini adalah bug, tetapi memperbaikinya sulit dilakukan dengan cara yang kompatibel dengan versi sebelumnya, terutama karena perataan.
Argumen:
- ruang lingkup: Objek Lingkup
- masukan: 4-D dengan bentuk
[batch, height, width, channels]
. Masukan untuk dikumpulkan. - ksize : Ukuran jendela untuk setiap dimensi tensor masukan.
- langkah: Langkah jendela geser untuk setiap dimensi tensor masukan.
- padding: Jenis algoritma padding yang akan digunakan.
Atribut opsional (lihat Attrs
):
- include_batch_in_index: Apakah akan menyertakan dimensi batch dalam indeks
argmax
yang diratakan.
Pengembalian:
- Keluaran
Output
: Tensor keluaran gabungan maksimal. - Argmaks
Output
: 4-D. Indeks rata dari nilai maksimal yang dipilih untuk setiap keluaran.
Konstruktor dan Destruktor | |
---|---|
MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding) | |
MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const MaxPoolWithArgmax::Attrs & attrs) |
Atribut publik | |
---|---|
argmax | |
operation | |
output |
Fungsi statis publik | |
---|---|
IncludeBatchInIndex (bool x) | |
Targmax (DataType x) |
Struktur | |
---|---|
tensorflow:: ops:: MaxPoolWithArgmax:: Attrs | Penyetel atribut opsional untuk MaxPoolWithArgmax . |
Atribut publik
argmax
::tensorflow::Output argmax
operasi
Operation operation
keluaran
::tensorflow::Output output
Fungsi publik
MaxPoolDenganArgmax
MaxPoolWithArgmax( const ::tensorflow::Scope & scope, ::tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding )
MaxPoolDenganArgmax
MaxPoolWithArgmax( const ::tensorflow::Scope & scope, ::tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const MaxPoolWithArgmax::Attrs & attrs )
Fungsi statis publik
SertakanBatchInIndex
Attrs IncludeBatchInIndex( bool x )
Targmaks
Attrs Targmax( DataType x )