gsnn.gsnn.models.GroupLayerNorm

Applies 1d layer normalization within each provided channel groups.

Classes

GroupLayerNorm(*args, **kwargs)

class gsnn.gsnn.models.GroupLayerNorm.GroupLayerNorm(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x)[source]