tensorflow::ops::FusedPadConv2D
#include <nn_ops.h>
Выполняет заливку как предподготовку во время свёртки.
Краткое описание
Подобно операции FusedResizeAndPadConv2d, эта операция позволяет оптимизировать реализацию, где этап пространственной заливкой интегрирован с поиске im2col, но в данном случае без билинейной фильтрации, необходимой для изменения размеров. Интеграция заливочного этапа предотвращает необходимость записи промежуточных результатов как полных тензоров, снижая нагрузку на память, и мы можем получить выигрыш в задержке, объединив вычисления преобразований. Атрибут data_format для Conv2D не поддерживается этой операцией, вместо этого используется порядок 'NHWC'. Внутренне эта операция использует один общий буфер на граф, что означает, что она будет блокировать, если выполняются несколько версий параллельно. Это связано с тем, что эта операция в первую очередь является оптимизацией для минимизации использования памяти.
Аргументы:
- scope: Объект Scope
- input: 4-мерный тензор с формой
[batch, in_height, in_width, in_channels]. - paddings: Матрица с двумя столбцами, задающая размеры заливочных добавлений. Количество строк должно быть таким же, как ранг
input. - filter: 4-мерный тензор с формой
[filter_height, filter_width, in_channels, out_channels]. - strides: 1-мерный массив длиной 4. Шаг скользящего окна для каждого измерения
input. Должен быть в том же порядке, что и размерность, указанная с формате. - padding: Тип алгоритма заполнения.
Возвращает:
-
Output: Тензор результата.
| Конструкторы и деструкторы | |
|---|---|
FusedPadConv2D(const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding) |
| Публичные атрибуты | |
|---|---|
operation | |
output | |
| Публичные функции | |
|---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const | |
Публичные атрибуты
операция
Operation operation
выход
::tensorflow::Output output
Публичные функции
FusedPadConv2D
FusedPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding )
узел
::tensorflow::Node * node() const
оператор::tensorflow::Вход
operator::tensorflow::Input() const
оператор::tensorflow::Выход
operator::tensorflow::Output() const
© 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/class/tensorflow/ops/fused-pad-conv2-d