首先,我使用numpy创建一个ndarray数组:
>>>a = np.array([1.0,2.0])
>>>a.dtye
dtype('float64')
这没什么问题,可以看出numpy默认创建的浮点数类型是float64
接下来,我将ndarray
数组转化成PyTroch的tensor
首先,我使用numpy创建一个ndarray数组:
>>>a = np.array([1.0,2.0])
>>>a.dtye
dtype('float64')
这没什么问题,可以看出numpy默认创建的浮点数类型是float64
接下来,我将ndarray
数组转化成PyTroch的tensor