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)