tf.raw_ops.Conv
Вычисляет N-мерную свёртку, заданную (N+1+batch_dims)-мерным input и (N+2)-мерным filter тензорами.
tf.raw_ops.Conv(
input,
filter,
strides,
padding,
explicit_paddings=[],
data_format='CHANNELS_LAST',
dilations=[],
batch_dims=1,
groups=1,
name=None
)
Общая функция для вычисления N-мерной свёртки. Требуется, чтобы 1 <= N <= 3.
| Аргументы | |
|---|---|
input | Tensor. Должен быть одного из следующих типов: half, bfloat16, float32, float64, int32. Тензор типа T и формы batch_shape + spatial_shape + [in_channels] в случае, если channels_last_format = true или формы batch_shape + [in_channels] + spatial_shape, если channels_last_format = false. spatial_shape имеет N-мерную размерность с N=2 или N=3. Обратите также внимание, что batch_shape определяется параметром batch_dims и по умолчанию равен 1. |
filter | Tensor. Должен иметь тот же тип, что и input. (N+2)-D тензор того же типа, что и input, и формой spatial_filter_shape + [in_channels, out_channels], где spatial_filter_shape имеет N-мерную размерность с N=2 или N=3. |
strides | Список ints. 1-мерный тензор длиной N+2. Шаг скользящего окна для каждой размерности input. Должен иметь strides[0] = strides[N+1] = 1. |
padding | string из: "SAME", "VALID", "EXPLICIT". Тип алгоритма заполнения, который следует использовать. |
explicit_paddings | Необязательный список ints. По умолчанию []. Если padding равно "EXPLICIT", список явных значений заполнения. Для i-й размерности количество вставляемых значений заполнения до и после размерности равно explicit_paddings[2 * i] и explicit_paddings[2 * i + 1] соответственно. Если padding не равно "EXPLICIT", explicit_paddings должно быть пустым. |
data_format | Необязательный string из: "CHANNELS_FIRST", "CHANNELS_LAST". По умолчанию "CHANNELS_LAST". Используется для установки формата данных. По умолчанию CHANNELS_FIRST, использует NHWC (2D) / NDHWC (3D) или если CHANNELS_LAST, использует NCHW (2D) / NCDHW (3D). |
dilations | Необязательный список ints. По умолчанию []. 1-мерный тензор длиной N+2. Коэффициент расширения для каждой размерности input. Если установлено значение k > 1, между каждым элементом фильтра в этой размерности будут пропущены k-1 ячеек. Порядок размерностей определяется значением channels_last_format, см. подробности выше. Разбавления в размерностях партии и глубины должны быть равны 1. |
batch_dims | Необязательный int. По умолчанию 1. Положительное целое число, определяющее количество размерностей партии для входного тензора. Должно быть меньше ранга входного тензора. |
groups | Необязательный int. По умолчанию 1. Положительное целое число, определяющее количество групп, на которые входной тензор разбивается по оси канала. Каждая группа свёртывается отдельно с filters / groups фильтрами. Выход представляет собой конкатенацию результатов всех групп по оси канала. Входные каналы и фильтры должны быть кратны группам. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Tensor. Имеет тот же тип, что и input. |
© 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/raw_ops/Conv