PerDimScale

A layer to scale individual dims of the input.

Abstract Signature:

PerDimScale(dim: int = 0)

PyTorch

API: —
Strategy: Custom / Partial

PaxML / Praxis

API: paxml.layers.PerDimScale
Strategy: Direct Mapping
Official Docs ↗