tf.keras.ops.depthwise_conv
Общая N-мерная глубинная свёртка.
tf.keras.ops.depthwise_conv(
inputs,
kernel,
strides=1,
padding='valid',
data_format=None,
dilation_rate=1
)
Эта операция поддерживает глубинную свёртку 1D и 2D.
| Аргументы | |
|---|---|
inputs | Тензор ранга N+2. inputs имеет форму (batch_size,) + inputs_spatial_shape + (num_channels,), если data_format="channels_last", или (batch_size, num_channels) + inputs_spatial_shape, если data_format="channels_first". |
kernel | Тензор ранга N+2. kernel имеет форму [kernel_spatial_shape, num_input_channels, num_channels_multiplier]; num_input_channels должно совпадать с количеством каналов в inputs. |
strides | Целое число или кортеж/список целых чисел размером len(inputs_spatial_shape), задающий шаг свёртки вдоль каждого пространственного измерения. Если strides — целое число, то для всех пространственных измерений используется одинаковое значение strides. |
padding | Строка: "valid" или "same". "valid" означает, что дополнение не применяется, а "same" равномерно добавляет дополнение слева/справа или сверху/снизу входных данных так, чтобы высота/ширина выходных данных совпадала с высотой/шириной входных данных, если strides=1. |
data_format | Строка: "channels_last" или "channels_first". data_format определяет порядок измерений во входных данных. Если data_format="channels_last", inputs имеет форму (batch_size, ..., channels), тогда как при data_format="channels_first" inputs имеет форму (batch_size, channels, ...). |
dilation_rate | Целое число или кортеж/список целых чисел размером len(inputs_spatial_shape), задающий коэффициент расширения для расширенной свёртки. Если dilation_rate — целое число, то для всех пространственных измерений используется одинаковое значение dilation_rate. |
| Возвращает | |
|---|---|
| Тензор ранга N+2 — результат операции глубинной свёртки. |
© 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/python/tf/keras/ops/depthwise_conv
