Class Splat2DModule

scripts/python/gaussian2d/gaussian2d.py:111–116

class Splat2DModule(torch.nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self, blobs):
        return Splat2DFunction.apply(blobs)


# create testbed instance with a window