Method forward

trellis/modules/sparse/norm.py:57–58

(self, x: SparseTensor)

    A LayerNorm layer that converts to float32 before the forward pass.
    """
    def forward(self, x: SparseTensor) -> SparseTensor:
        return super().forward(x.float()).type(x.dtype)