tensorflow::ops::Conv2D::Attrs
#include <nn_ops.h>
Дополнительные атрибуты-сеттеры для Conv2D.
Summary
| Общедоступные атрибуты | |
|---|---|
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
Общедоступные функции
DataFormat
TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::DataFormat( StringPiece x )
Указывает формат данных входных и выходных данных.
С использованием формата по умолчанию "NHWC", данные хранятся в порядке: [batch, высота, ширина, каналы]. В качестве альтернативы, формат может быть "NCHW", порядок хранения данных: [batch, каналы, высота, ширина].
По умолчанию "NHWC"
Dilations
TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::Dilations( const gtl::ArraySlice< int > & x )
1-мерный тензор длиной 4.
Коэффициент дилатации для каждого измерения input. Если установлено значение k > 1, между каждым элементом фильтра в этом измерении будет k-1 пропущенных ячеек. Порядок измерений определяется значением data_format, см. подробности выше. Дилатации в измерениях пакет и глубина должны быть 1.
По умолчанию [1, 1, 1, 1]
ExplicitPaddings
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 должно быть пустым.
По умолчанию []
UseCudnnOnGpu
TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::UseCudnnOnGpu( bool x )
По умолчанию true.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/cc/struct/tensorflow/ops/conv2-d/attrs