aliran tensor:: operasi:: FusedPadConv2D

#include <nn_ops.h>

Melakukan padding sebagai praproses selama konvolusi.

Ringkasan

Mirip dengan FusedResizeAndPadConv2d, operasi ini memungkinkan implementasi yang dioptimalkan di mana tahap transformasi padding spasial digabungkan dengan pencarian im2col, tetapi dalam kasus ini tanpa pemfilteran bilinear yang diperlukan untuk mengubah ukuran. Penggabungan padding mencegah kebutuhan untuk menuliskan hasil antara sebagai tensor keseluruhan, sehingga mengurangi tekanan memori, dan kita bisa mendapatkan beberapa peningkatan latensi dengan menggabungkan perhitungan transformasi. Atribut data_format untuk Conv2D tidak didukung oleh operasi ini, dan sebagai gantinya digunakan urutan 'NHWC'. Secara internal, operasi ini menggunakan buffer awal per grafik, yang berarti operasi ini akan diblokir jika beberapa versi dijalankan secara paralel. Pasalnya, operator ini pada dasarnya melakukan optimasi untuk meminimalkan penggunaan memori.

Argumen:

  • ruang lingkup: Objek Lingkup
  • masukan: 4-D dengan bentuk [batch, in_height, in_width, in_channels] .
  • paddings: Matriks dua kolom yang menentukan ukuran padding. Jumlah baris harus sama dengan pangkat yang input .
  • filter: 4-D dengan bentuk [filter_height, filter_width, in_channels, out_channels] .
  • langkah: 1-D panjang 4. Langkah jendela geser untuk setiap dimensi input . Harus berada dalam urutan yang sama dengan dimensi yang ditentukan dengan format.
  • padding: Jenis algoritma padding yang akan digunakan.

Pengembalian:

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::Masukan

 operator::tensorflow::Input() const 

operator::tensorflow::Keluaran

 operator::tensorflow::Output() const