Формы, шаг, дополнение и прореживание

Расчёт формы управляет адресацией

Шаг задаёт расстояние между началами выходных окон. Прореживание задаёт расстояние между отсчётами внутри ядра. Они пропускают входные позиции в разных членах индекса. Ядро из R отсчётов с прореживанием D охватывает D(R−1)+1 позиций. Задавайте верхнее, нижнее, левое и правое дополнения независимо. Упражнения используют явное дополнение и выходные размеры с округлением вниз, а не подразумевают автоматическое соглашение SAME.

Для выходной ширины замените высоту и вертикальные параметры горизонтальными. Наша учебная модель возвращает ноль выходных позиций, когда эффективное ядро превышает дополненный вход. Производственный оператор должен отдельно определить, принимает ли он пустой тензор.

Разобранные примеры

При H=5, R=3, шаге 2, верхнем и нижнем дополнении один и прореживании один эффективная высота равна трём. Выходная высота — floor((5+2−3)/2)+1=3. Изменение прореживания на два делает эффективную высоту пять и даёт выходную высоту два. Шаг не менялся; каждое окно теперь охватывает большую область.

Отображайте выходную координату и координату ядра через ih=h×S−Pt+r×D. При h=1,r=2 первая конфигурация читает 3. Конфигурация с прореживанием читает 5. Допустимые входные координаты — от нуля до четырёх, поэтому координата пять подаёт дополнение, а не читает за массивом. Дополнение представляет действительный ноль; сырое квантованное хранение может кодировать этот ноль ненулевым кодом.

Линейный адрес NCHW — ((nC+c)H+ih)W+iw. При N=1,C=2,H=W=3 координата (0,1,2,1) отображается в 16 в тензоре из 18 элементов. Это средний отсчёт последней строки второго канала. Перестановка канала и пространственных осей сохраняет число элементов, но меняет вычисление.

При G группах числа входных и выходных каналов должны делиться на G. Каждый выход сворачивает C/G входных каналов своей группы. Вычисляйте выходные размеры, генерируйте адреса и проверяйте каждую входную границу именно в этом порядке. Официальная спецификация ONNX Conv определяет соответствующие атрибуты явного дополнения, шага, прореживания и групп.