tensorflow::ops::Conv2D::Attrs

#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` 1-мерный тензор длиной 4.
`ExplicitPaddings(const gtl::ArraySlice< int > & x)`	`TF_MUST_USE_RESULT Attrs` Если `padding` равно `"EXPLICIT"`, список явных значений заполнения.
`UseCudnnOnGpu(bool x)`	`TF_MUST_USE_RESULT Attrs` По умолчанию true.

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

data_format_

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

dilations_

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

explicit_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", данные хранятся в порядке: [batch, height, width, channels]. В качестве альтернативы, можно использовать формат "NCHW", в котором порядок хранения данных: [batch, channels, height, width].

По умолчанию "NHWC"

Растяжения

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

1-мерный тензор длиной 4.

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

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

Использовать Cudnn на GPU

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

По умолчанию true.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/cc/struct/tensorflow/ops/conv2-d/attrs