Как установить верхнюю границу для нормы L2 в PyTorch

Question

Я пытаюсь переопределить в PyTorch CNN, который изначально был реализован в Lua / Torch. Он использует Module:maxParamNorm([maxOutNorm, maxInNorm]) из dpnn: глубокие расширения для nn , чтобы установить верхнюю границу для нормы L2 весовых векторов (согласно p2 из Улучшение нейронных сетей путем предотвращения совместной адаптации детекторов признаков ).

Как мне это сделать в PyTorch?

Я просмотрел документы и не нашел ничего похожего на это, но я новичок в PyTorch ie так что, возможно, что-то упустил.