PyTorch torch.ones([])