tensor akışı:: işlem:: FusedPadConv2D

#include <nn_ops.h>

Evrişim sırasında ön işlem olarak dolgu gerçekleştirir.

Özet

FusedResizeAndPadConv2d'ye benzer şekilde, bu işlem, uzamsal dolgu dönüştürme aşamasının im2col aramasıyla birleştirildiği, ancak bu durumda yeniden boyutlandırma için gereken çift doğrusal filtrelemenin olmadığı optimize edilmiş bir uygulamaya izin verir. Dolguyu birleştirmek, ara sonuçları tam tensörler olarak yazma ihtiyacını ortadan kaldırarak hafıza basıncını azaltır ve dönüşüm hesaplamalarını birleştirerek bazı gecikme kazanımları elde edebiliriz. Conv2D'nin data_format özelliği bu operasyon tarafından desteklenmiyor ve bunun yerine 'NHWC' sırası kullanılıyor. Dahili olarak bu operasyon, grafik başına tek bir karalama arabelleği kullanır; bu, birden fazla sürümün paralel olarak çalıştırılması durumunda engelleneceği anlamına gelir. Bunun nedeni, bu operatörün öncelikle bellek kullanımını en aza indirmeye yönelik bir optimizasyon olmasıdır.

Argümanlar:

  • kapsam: Bir Kapsam nesnesi
  • giriş: şekilli 4-D [batch, in_height, in_width, in_channels] .
  • dolgular: Dolgu boyutlarını belirten iki sütunlu bir matris. Satır sayısı input sıralamasıyla aynı olmalıdır.
  • filtre: 4-D şekilli [filter_height, filter_width, in_channels, out_channels] .
  • adımlar: 1 boyutlu uzunluk 4. input her boyutu için kayan pencerenin adımı. Formatla belirtilen boyutla aynı sırada olmalıdır.
  • padding: Kullanılacak dolgu algoritmasının türü.

İadeler:

  • Output : Çıkış tensörü.

Yapıcılar ve Yıkıcılar

FusedPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding)

Genel özellikler

operation
output

Kamu işlevleri

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Genel özellikler

operasyon

Operation operation

çıktı

::tensorflow::Output output

Kamu işlevleri

FusedPadConv2D

 FusedPadConv2D(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input paddings,
  ::tensorflow::Input filter,
  StringPiece mode,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

düğüm

::tensorflow::Node * node() const 

operatör::tensorflow::Giriş

 operator::tensorflow::Input() const 

operatör::tensorflow::Çıktı

 operator::tensorflow::Output() const