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

tensorflow :: ops :: Dönş2D

#include <nn_ops.h>

4-D input ve filter tensörleri verilen 2-D evrişimi hesaplar.

Özet

Bir şekil giriş tensörü [batch, in_height, in_width, in_channels] ve bir filtre / çekirdek tensörü [filter_height, filter_width, in_channels, out_channels] , bu işlem [filter_height, filter_width, in_channels, out_channels] gerçekleştirir:

  1. Filtreyi şekil [filter_height * filter_width * in_channels, output_channels] ile 2 boyutlu bir matrise [filter_height * filter_width * in_channels, output_channels] .
  2. Sanal bir şekil tensörü [batch, out_height, out_width, filter_height * filter_width * in_channels] oluşturmak için giriş tensöründen görüntü yamalarını [batch, out_height, out_width, filter_height * filter_width * in_channels] .
  3. Her yama için filtre matrisini ve görüntü yaması vektörünü sağa doğru çarpar.

Ayrıntılı olarak, varsayılan NHWC formatı

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]
ile

strides[0] = strides[3] = 1 . Aynı yatay ve tepe adımlarının en yaygın durumu için strides = [1, stride, stride, 1] .

Argümanlar:

  • kapsam: Bir Scope nesnesi
  • giriş: 4-D tensör. Boyut sırası data_format değerine göre yorumlanır, ayrıntılar için aşağıya bakın.
  • filtre: 4 boyutlu bir şekil tensörü [filter_height, filter_width, in_channels, out_channels]
  • adım sayısı: 1-D uzunluk tensörü 4. input her boyutu için kayan pencerenin adım sayısı. Boyut sırası data_format değerine göre data_format , 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 ):

  • explicit_paddings: Eğer padding "EXPLICIT" , açık dolgu miktarlarının listesi. explicit_paddings[2 * i] . Boyut için, boyuttan önce ve sonra eklenen dolgu miktarı sırasıyla explicit_paddings[2 * i] ve explicit_paddings[2 * i + 1] . padding "EXPLICIT" değilse, explicit_paddings boş olmalıdır.
  • data_format: Giriş ve çıkış verilerinin veri formatını belirtin. Varsayılan biçim "NHWC" ile veriler şu sırayla saklanır: [toplu iş, yükseklik, genişlik, kanallar]. Alternatif olarak, format "NCHW" olabilir, veri saklama sırası: [parti, kanallar, yükseklik, genişlik].
  • genişlemeler: 1-D uzunluk tensörü 4. Her input boyutu için genişleme faktörü. K> 1 olarak ayarlanırsa, o boyuttaki her filtre öğesi arasında k-1 atlanan hücreler olacaktır. Boyut sırası data_format değerine göre data_format , ayrıntılar için yukarıya bakın. Parti ve derinlik boyutlarındaki genişlemeler 1 olmalıdır.

İadeler:

  • Output : 4-D tensör. Boyut sırası data_format değerine göre data_format , 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

Kamusal işlevler

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

Dönüşüm2D için isteğe bağlı öznitelik belirleyiciler.

Genel özellikler

operasyon

Operation operation

çıktı

::tensorflow::Output output

Kamusal işlevler

Dönş2D

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

Dönş2D

 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 

operator :: tensorflow :: Girdi

 operator::tensorflow::Input() const 

operator :: tensorflow :: Çıktı

 operator::tensorflow::Output() const 

Genel statik işlevler

Veri formatı

Attrs DataFormat(
  StringPiece x
)

Dilatasyonlar

Attrs Dilations(
  const gtl::ArraySlice< int > & x
)

ExplicitPaddings

Attrs ExplicitPaddings(
  const gtl::ArraySlice< int > & x
)

UseCudnnOnGpu

Attrs UseCudnnOnGpu(
  bool x
)