CrossMapLRN

Описание

Info

Родительский класс: LRN

Производные классы: -

Этот модуль реализует операцию межканальной локальной нормализации.

Инициализация

def __init__(self, N=5, alpha=1e-4, beta=0.75, K=2.0, name=None):

Параметры

Параметр Возможные типы Описание По умолчанию
N int Количество каналов для суммирования. 5
alpha float Сдвиг. 1e-4
beta float Параметр масштаба. 0.75
K float Добавочный коэффициент. 2.0
name str Имя слоя. None

Пояснения

-

Примеры

Необходимые импорты.

>>> import numpy as np
>>> from PuzzleLib.Backend import gpuarray
>>> from PuzzleLib.Modules import CrossMapLRN

Info

gpuarray необходим для правильного размещения тензора на GPU

>>> batchsize, maps, h, w = 1, 10, 2, 2
>>> data = gpuarray.to_gpu(np.random.randn(batchsize, maps, h, w).astype(np.float32))
>>> lrn = CrossMapLRN()
>>> lrn(data)