ncdl.lattice.LatticeTensor.coset

LatticeTensor.coset(coset: int) → torch.Tensor

Returns the underlying tensor for a given coset index. Gradients are tracked for these tensors.

Returns:

A torch tensor on self.device