Method __init__

sat/sgm/modules/autoencoding/magvit2_pytorch.py:201–203

(self, fn: Module)

class ToTimeSequence(Module):
    @beartype
    def __init__(self, fn: Module):
        super().__init__()
        self.fn = fn

    def forward(self, x, **kwargs):
        x = rearrange(x, "b c f ... -> b ... f c")