is_tensor
torch.is_tensor(obj)
参数:obj
为对象。
作用:判断 obj
是否为 Tensor
对象。与 isinstance(obj, Tensor)
的作用相似,但是更推荐使用 isintsance
。
tensor
torch.tensor(data, dtype=None, device=None, requires_grad=False)
主要参数:
date
:可以是list
、tuple
,ndarray
,scaler
,或者是其他类型。dtype
:函数返回的张量的元素类型。可选值为torch.dtype
类型,比如torch.int
、torch.float
、torch.float64
等。device
:保存张量的设备。可选值为torch.device
类型。当device=None
时,如果data
是张量,那么返回张量的device
是data
所在设备;如果data
不是张量,那么返回张量的