Searched refs:SubclassWithTensorFactory (Results 1 – 2 of 2) sorted by relevance
222 class SubclassWithTensorFactory(torch.Tensor): class
6014 from torch.testing._internal.common_subclass import SubclassWithTensorFactory6016 inp = SubclassWithTensorFactory(torch.zeros(3, 5))