i-machine-think/diagNNose

Look into subclassing from torch.Tensor directly

jumelet opened this issue 4 years ago · 0 comments

jumelet commented 4 years ago

New functionality has been added in torch 1.7: https://pytorch.org/docs/stable/notes/extending.html#subclassing-torch-tensor