TypeError: expected CPU (got CUDA) when sublcassing / inheriting torch.Tensor class

[turian]

Solution, according to the issue:

@turian

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 = torch.zeros(batch_size, N, device='cuda').as_subclass(Signal)