tensorflow::ops::Conv2D::Атрибуты
#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
Общедоступные функции
DataFormat
TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::DataFormat( StringPiece x )
Указывает формат данных входных и выходных данных.
При использовании формата "NHWC" по умолчанию данные хранятся в порядке: [batch, height, width, channels]. В качестве альтернативы можно использовать формат "NCHW", порядок хранения данных: [batch, channels, height, width].
По умолчанию "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, подробности см. выше. Расширения по измерениям batch и depth должны быть равны 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 должно быть пустым.
По умолчанию []
Использовать 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/api_docs/cc/struct/tensorflow/ops/conv2-d/attrs