Source code for torchsparse.nn.modules.activation

from torch import nn

from torchsparse import SparseTensor
from torchsparse.nn.utils import fapply

__all__ = ['ReLU', 'LeakyReLU']


[docs]class ReLU(nn.ReLU):
    """ ReLU activation function. """

[docs]    def forward(self, input: SparseTensor) -> SparseTensor:
        return fapply(input, super().forward)


[docs]class LeakyReLU(nn.LeakyReLU):
    """ LeakyReLU activation function. """

[docs]    def forward(self, input: SparseTensor) -> SparseTensor:
        return fapply(input, super().forward)