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)