掌握Pytorch
学习内容:
Tensor的创建方式有四种,分别是torch.Tensor()、torch.tensor()、torch.as_tensor()、torch.from_numpy()。
1、torch.Tensor()
torch.Tensor()为类构造函数,创建后,默认dtype为float32。
data = np.ones([2,2],dtype = np.int64) # [[1 1] # [1 1]] #采用torch.Tensor()创建一个tensor torch.Tensor(data) # tensor([[1., 1.], # [1., 1.]])
2、torch.tensor()
torch.tensor()为工厂函数。工厂函数可以显式的设置数据类型,但是构造类函数不能直接设置。
#采用torch.tensor()创建一个tensor torch.tensor(data,dtype = torch.float16) # tensor([[1., 1.], # [1., 1.]], dtype=torch.float16)
3、torch.as_tensor()
torch.as_tensor()为工厂函数。
#采用torch.tensor()创建一个tensor torch.as_tensor(data) # tensor([[1, 1], # [1, 1]])
4、torch.from_numpy()
torch.from_numpy()为工厂函数。
#采用orch.from_numpy()创建一个tensor torch.from_numpy(data) # tensor([[1, 1], # [1, 1]])
Note:类构造函数与工厂函数的区别在于数据类型不同,工厂函数会根据输入推断数据类型。
总结:
要坚持要加油呀~



