Searched refs:make_scalar_tensor (Results 1 – 2 of 2) sorted by relevance
126 def make_scalar_tensor(dtype): function152 s = make_scalar_tensor(dtype=torch.chalf)176 s = make_scalar_tensor(dtype=torch.cfloat)195 s = make_scalar_tensor(dtype=torch.cdouble)
6368 …make_scalar_tensor = partial(make_tensor, (), device='cpu', dtype=dtype, requires_grad=requires_gr…6394 ((3, 4), make_scalar_tensor(), make_scalar_tensor()),