Searched refs:OptionalScaledTensor (Results 1 – 1 of 1) sorted by relevance
265 class OptionalScaledTensor(torch.Tensor): class298 return OptionalScaledTensor(310 return OptionalScaledTensor(1336 sub1 = OptionalScaledTensor(torch.randn(2, 4), None)1337 sub2 = OptionalScaledTensor(torch.randn(2, 4), torch.randn(2, 4))