
462 Глава 14
где n — пространственное измерение входной карты признаков, а p, m и s обозначают
заполнение, размер ядра и шаг соответственно. Следовательно, чтобы добиться o = i,
необходимо использовать p = 2.
Слои объединения по максимуму с размером ядра 2×2 и шагом 2 уменьшат пространст-
венные размеры вдвое. (Обратите внимание, что если параметр
stride
не указан
в
MaxPool2D
, по умолчанию он устанавливается равным размеру ядра.)
Хотя на этом этапе мы можем рассчитать размер карт признаков вручную, PyTorch
предоставляет нам удобный метод для расчета:
>>> x = torch.ones((4, 1, 28, 28))
>>> model(x).shape
torch.Size([4, 64, 7, ...