torch.get_default_device

torch.get_default_device()[源代码]

获取默认分配在device上的torch.Tensor

返回类型

设备