тензорный поток:: опс:: Конв2Д:: Атрибуты

#include <nn_ops.h>

Дополнительные установщики атрибутов для Conv2D .

Краткое содержание

Публичные атрибуты
data_format_ = "NHWC"	StringPiece
dilations_ = Default_dilations()	gtl::ArraySlice< int >
explicit_paddings_ = {}	gtl::ArraySlice< int >
use_cudnn_on_gpu_ = true	bool

Общественные функции
DataFormat (StringPiece x)	TF_MUST_USE_RESULT Attrs Укажите формат входных и выходных данных.
Dilations (const gtl::ArraySlice< int > & x)	TF_MUST_USE_RESULT Attrs 1D тензор длины 4.
ExplicitPaddings (const gtl::ArraySlice< int > & x)	TF_MUST_USE_RESULT Attrs Если padding имеет значение "EXPLICIT" , список явных сумм заполнения.
UseCudnnOnGpu (bool x)	TF_MUST_USE_RESULT Attrs По умолчанию истинно.

Публичные атрибуты

data_format_

StringPiece tensorflow::ops::Conv2D::Attrs::data_format_ = "NHWC"

расширения_

gtl::ArraySlice< int > tensorflow::ops::Conv2D::Attrs::dilations_ = Default_dilations()

явное_paddings_

gtl::ArraySlice< int > tensorflow::ops::Conv2D::Attrs::explicit_paddings_ = {}

use_cudnn_on_gpu_

bool tensorflow::ops::Conv2D::Attrs::use_cudnn_on_gpu_ = true

Общественные функции

Формат данных

TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::DataFormat(
  StringPiece x
)

Укажите формат входных и выходных данных.

В формате по умолчанию «NHWC» данные сохраняются в следующем порядке: [пакет, высота, ширина, каналы]. Альтернативно, формат может быть «NCHW», порядок хранения данных: [пакет, каналы, высота, ширина].

По умолчанию «NHWC».

Расширения

TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::Dilations(
  const gtl::ArraySlice< int > & x
)

1D тензор длины 4.

Коэффициент расширения для каждого измерения input . Если установлено значение k > 1, между каждым фильтрующим элементом в этом измерении будет k-1 пропущенных ячеек. Порядок измерений определяется значением data_format , подробности см. выше. Расширения размеров партии и глубины должны быть равны 1.

По умолчанию [1, 1, 1, 1]

Явные отступы

TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::ExplicitPaddings(
  const gtl::ArraySlice< int > & x
)

Если padding имеет значение "EXPLICIT" , список явных сумм заполнения.

Для i-го измерения величина заполнения, вставленная до и после измерения, равна explicit_paddings[2 * i] и explicit_paddings[2 * i + 1] соответственно. Если padding не "EXPLICIT" , explicit_paddings должно быть пустым.

По умолчанию []

Усекудннонгпу

TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::UseCudnnOnGpu(
  bool x
)

По умолчанию истинно.