tf.contrib.layers.conv2d
Добавляет N-мерную свёртку, за которой следует необязательный слой batch_norm.
tf.contrib.layers.conv2d(
inputs, num_outputs, kernel_size, stride=1, padding='SAME', data_format=None,
rate=1, activation_fn=tf.nn.relu, normalizer_fn=None, normalizer_params=None,
weights_initializer=initializers.xavier_initializer(), weights_regularizer=None,
biases_initializer=tf.zeros_initializer(), biases_regularizer=None, reuse=None,
variables_collections=None, outputs_collections=None, trainable=True, scope=None
)
Требуется, чтобы 1 <= N <= 3.
convolution создаёт переменную с именем weights, представляющую свёрточный ядро, которое свёртывается (на самом деле взаимокоррелируется) с inputs, чтобы получить Tensor активаций. Если указана функция normalizer_fn (например, batch_norm), она применяется. В противном случае, если normalizer_fn равно None и указана функция biases_initializer, то создаётся переменная biases и добавляется к активациям. Наконец, если activation_fn не равно None, она также применяется к активациям.
Выполняет атросную свёртку со скоростью входного шага/разрежением равной rate, если для любого измерения rate указано значение > 1. В этом случае значения stride != 1 не поддерживаются.
| Аргументы | |
|---|---|
inputs | Tensor ранга N+2 с формой [batch_size] + input_spatial_shape + [in_channels] если data_format не начинается с "NC" (по умолчанию) или [batch_size, in_channels] + input_spatial_shape если data_format начинается с "NC". |
num_outputs | Целое число, количество выходных фильтров. |
kernel_size | Последовательность из N положительных целых чисел, определяющих пространственные размеры фильтров. Может быть одним целым числом, чтобы указать одинаковое значение для всех пространственных измерений. |
stride | Последовательность из N положительных целых чисел, определяющих шаг для вычисления выходных данных. Может быть одним целым числом, чтобы указать одинаковое значение для всех пространственных измерений. Указание любого значения stride != 1 несовместимо с указанием любого значения rate != 1. |
padding | Один из "VALID" или "SAME". |
data_format | Строка или None. Указывает, является ли измерение канала input и выхода последним измерением (по умолчанию, или если data_format не начинается с "NC") или вторым измерением (если data_format начинается с "NC"). Для N=1 допустимые значения - "NWC" (по умолчанию) и "NCW". Для N=2 допустимые значения - "NHWC" (по умолчанию) и "NCHW". Для N=3 допустимые значения - "NDHWC" (по умолчанию) и "NCDHW". |
rate | Последовательность из N положительных целых чисел, определяющих скорость разрежения для атросной свёртки. Может быть одним целым числом, чтобы указать одинаковое значение для всех пространственных измерений. Указание любого значения rate != 1 несовместимо с указанием любого значения stride != 1. |
activation_fn | Функция активации. Значение по умолчанию - функция ReLU. Явно установите её в None, чтобы пропустить её и сохранить линейную активацию. |
normalizer_fn | Функция нормализации для использования вместо biases. Если normalizer_fn предоставлена, то biases_initializer и biases_regularizer игнорируются, и biases не создаются и не добавляются. По умолчанию установлено в None для отсутствия функции нормализации. |
normalizer_params | Параметры функции нормализации. |
weights_initializer | Инициализатор весов. |
weights_regularizer | Необязательный регуляризатор для весов. |
biases_initializer | Инициализатор смещений. Если None, смещения пропускаются. |
biases_regularizer | Необязательный регуляризатор для смещений. |
reuse | Нужно ли повторно использовать слой и его переменные. Для возможности повторного использования необходимо указать область действия слоя. |
variables_collections | Необязательный список коллекций для всех переменных или словарь, содержащий различные списки коллекций для каждой переменной. |
outputs_collections | Коллекция для добавления выходных данных. |
trainable | Если True также добавить переменные в коллекцию графа GraphKeys.TRAINABLE_VARIABLES (см. tf.Variable). |
scope | Необязательная область действия для variable_scope. |
conv_dims | Необязательная размерность свёртки, при установке которой будет использоваться соответствующая свёртка (например, 2 для Conv 2D, 3 для Conv 3D и т.д.). Если оставлено равным None, то размерность свёртки будет выбрана на основе ранга входных данных (т.е. Conv ND, с N = input_rank - 2). |
| Возвращаемые значения | |
|---|---|
| Tensor, представляющий выход операции. |
| Исключения | |
|---|---|
ValueError | Если data_format некорректен. |
ValueError | И 'rate', и stride не равны 1. |
© 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/r1.15/api_docs/python/tf/contrib/layers/conv2d