Searched refs:TorchDispatchTensor (Results 1 – 1 of 1) sorted by relevance
60 class TorchDispatchTensor(torch.Tensor): class70 return x.elem if isinstance(x, TorchDispatchTensor) else x73 return TorchDispatchTensor(x) if isinstance(x, torch.Tensor) else x728 x = TorchDispatchTensor(torch.ones((1,)))