ML Topluluk Günü 9 Kasım! TensorFlow, JAX güncellemeler için bize katılın ve daha fazla bilgi edinin

tensorflow :: ops :: FusedPadConv2D

#include <nn_ops.h>

Bir evrişim sırasında ön işlem olarak bir 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ırır, bellek baskısını azaltır ve dönüşüm hesaplamalarını birleştirerek bazı gecikme kazançları elde edebiliriz. İçin data_format nitelik Conv2D bu op tarafından desteklenmemektedir ve 'NHWC' sipariş yerine kullanılır. Dahili olarak bu operasyon, grafik başına tek bir çalışma tamponu 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 Scope nesnesi
  • girdi: [batch, in_height, in_width, in_channels] şeklinde 4-D.
  • paddings: Dolgu boyutlarını belirten iki sütunlu bir matris. Satırların sayısı, input sıralamasıyla aynı olmalıdır.
  • filtre: 4 boyutlu [filter_height, filter_width, in_channels, out_channels] .
  • adım sayısı: 1-D uzunluğunda 4. Her input boyutu için kayan pencerenin adım sayısı. Format ile belirtilen boyut ile 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

Kamusal işlevler

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

Genel özellikler

operasyon

Operation operation

çıktı

::tensorflow::Output output

Kamusal işlevler

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 

operator :: tensorflow :: Girdi

 operator::tensorflow::Input() const 

operator :: tensorflow :: Çıktı

 operator::tensorflow::Output() const