tensorflow::ops::FusedPadConv2D
#include <nn_ops.h>Melakukan padding sebagai pra-pemrosesan selama konvolusi.
Ringkasan
Mirip dengan FusedResizeAndPadConv2d, operasi ini memungkinkan implementasi yang dioptimalkan di mana tahap transformasi padding spasial digabungkan dengan pencarian im2col, tetapi dalam hal ini tanpa pemfilteran bilinear yang diperlukan untuk pengubahan ukuran. Penggabungan padding mencegah perlunya menulis hasil perantara sebagai tensor utuh, mengurangi tekanan memori, dan kita dapat memperoleh beberapa keuntungan latensi dengan menggabungkan perhitungan transformasi. Atribut data_format untuk Conv2D tidak didukung oleh operasi ini, dan urutan 'NHWC' digunakan sebagai gantinya. Secara internal, operasi ini menggunakan satu buffer sementara per-grafik, yang berarti akan terblokir jika beberapa versi dijalankan secara paralel. Hal ini karena operator ini terutama merupakan optimasi untuk meminimalkan penggunaan memori.
Argumen:
- Cakupan: Objek Cakupan
- input: 4-D dengan bentuk
[batch, in_height, in_width, in_channels]. - paddings: Matriks dua kolom yang menentukan ukuran padding. Jumlah baris harus sama dengan rank
input. - filter: 4-D dengan bentuk
[filter_height, filter_width, in_channels, out_channels]. - strides: 1-D dengan panjang 4. Langkah jendela geser untuk setiap dimensi
input. Harus dalam urutan yang sama dengan dimensi yang ditentukan dengan format. - padding: Jenis algoritma padding yang akan digunakan.
Mengembalikan:
-
Output: Tensor output.
Konstruktor dan Destruktor | |
|---|---|
FusedPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding) |
Atribut publik | |
|---|---|
operation | |
output | |
Fungsi publik | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Atribut publik
operasi
Operation operation
keluaran
::tensorflow::Output output
Fungsi publik
FusedPadConv2D
FusedPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding )
simpul
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const