Here is the issue 15. We have the same issue.
What is the fix we should apply?
import torch

class Signal(torch.Tensor):
    @property
    def batch_size(self):
        assert self.ndim == 2
        return self.shape[0]

    @property
    def num_samples(self):
        assert self.ndim == 2
        return self.shape[1]

N = 88200
batch_size = 64
pitch = Signal(torch.zeros(batch_size, N, device='cuda'))

TypeError: expected CPU (got CUDA)

What should we be doing instead?