tensor akışı:: işlem:: Dönüşüm2D
#include <nn_ops.h>
4 boyutlu input
ve filter
tensörleri verildiğinde 2 boyutlu bir evrişimi hesaplar.
Özet
Şekil [batch, in_height, in_width, in_channels]
şeklinde bir giriş tensörü ve [filter_height, filter_width, in_channels, out_channels]
şeklinde bir filtre/çekirdek tensörü verildiğinde, bu op aşağıdakileri gerçekleştirir:
- Filtreyi
[filter_height * filter_width * in_channels, output_channels]
şeklinde 2 boyutlu bir matrise düzleştirir. -
[batch, out_height, out_width, filter_height * filter_width * in_channels]
şeklinde sanal bir tensör oluşturmak için giriş tensöründen görüntü yamalarını çıkarır. - Her yama için filtre matrisini ve görüntü yama vektörünü sağla çarpar.
Ayrıntılı olarak, varsayılan NHWC formatıyla,
output[b, i, j, k] = sum_{di, dj, q} input[b, strides[1] * i + di, strides[2] * j + dj, q] * filter[di, dj, q, k]
strides[0] = strides[3] = 1
olmalıdır. Aynı yatay ve köşeli adımların en yaygın durumu için, strides = [1, stride, stride, 1]
.
Argümanlar:
- kapsam: Bir Kapsam nesnesi
- giriş: 4 boyutlu bir tensör. Boyut sırası
data_format
değerine göre yorumlanır; ayrıntılar için aşağıya bakın. - filtre: Şeklin 4 boyutlu tensörü
[filter_height, filter_width, in_channels, out_channels]
- adımlar: uzunluğun 1 boyutlu tensörü 4.
input
her boyutu için kayan pencerenin adımı. Boyut sırasıdata_format
değerine göre belirlenir; ayrıntılar için aşağıya bakın. - padding: Kullanılacak dolgu algoritmasının türü.
İsteğe bağlı özellikler (bkz. Attrs
):
- açık_paddingler:
padding
"EXPLICIT"
ise, açık dolgu miktarlarının listesi. i. boyut için, boyuttan önce ve sonra eklenen dolgu miktarı sırasıylaexplicit_paddings[2 * i]
veexplicit_paddings[2 * i + 1]
şeklindedir.padding
"EXPLICIT"
değilse,explicit_paddings
boş olmalıdır. - data_format: Giriş ve çıkış verilerinin veri formatını belirtin. Varsayılan format "NHWC" ile veriler şu sırayla saklanır: [toplu iş, yükseklik, genişlik, kanallar]. Alternatif olarak format, veri depolama sırası olan "NCHW" olabilir: [toplu iş, kanallar, yükseklik, genişlik].
- genişlemeler: 1-D uzunluk tensörü 4.
input
her boyutu için genişleme faktörü. k > 1 olarak ayarlanırsa, o boyuttaki her filtre elemanı arasında k-1 atlanan hücre olacaktır. Boyut sırasıdata_format
değerine göre belirlenir; ayrıntılar için yukarıya bakın. Parti ve derinlik boyutlarındaki genişlemeler 1 olmalıdır.
İade:
-
Output
: 4 boyutlu bir tensör. Boyut sırasıdata_format
değerine göre belirlenir; ayrıntılar için aşağıya bakın.
Yapıcılar ve Yıkıcılar | |
---|---|
Conv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input filter, const gtl::ArraySlice< int > & strides, StringPiece padding) | |
Conv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input filter, const gtl::ArraySlice< int > & strides, StringPiece padding, const Conv2D::Attrs & attrs) |
Genel özellikler | |
---|---|
operation | |
output |
Kamu işlevleri | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Genel statik işlevler | |
---|---|
DataFormat (StringPiece x) | |
Dilations (const gtl::ArraySlice< int > & x) | |
ExplicitPaddings (const gtl::ArraySlice< int > & x) | |
UseCudnnOnGpu (bool x) |
Yapılar | |
---|---|
tensorflow:: ops:: Conv2D:: Öznitelikler | Conv2D için isteğe bağlı özellik ayarlayıcılar. |
Genel özellikler
operasyon
Operation operation
çıktı
::tensorflow::Output output
Kamu işlevleri
Dönüşüm2D
Conv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input filter, const gtl::ArraySlice< int > & strides, StringPiece padding )
Dönüşüm2D
Conv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input filter, const gtl::ArraySlice< int > & strides, StringPiece padding, const Conv2D::Attrs & attrs )
düğüm
::tensorflow::Node * node() const
operatör::tensorflow::Giriş
operator::tensorflow::Input() const
operatör::tensorflow::Çıktı
operator::tensorflow::Output() const
Genel statik işlevler
Veri Formatı
Attrs DataFormat( StringPiece x )
Dilatasyonlar
Attrs Dilations( const gtl::ArraySlice< int > & x )
Açık Dolgular
Attrs ExplicitPaddings( const gtl::ArraySlice< int > & x )
CudnnOnGpu'yu kullanın
Attrs UseCudnnOnGpu( bool x )