spacetimeformer.spacetimeformer_model.utils package

class spacetimeformer.spacetimeformer_model.utils.masking.ProbMask(B, H, L, index, scores, device='cpu')[source]

Bases: object

class spacetimeformer.spacetimeformer_model.utils.masking.TriangularCausalMask(B, L, device='cpu')[source]

Bases: object