pytorch基础入门day01

对pytorch的张量创建：

#张量:与numpy相似(tensor) # 分为维度,形状,数据类型 # 张量的创建 import torch # 创建一个2*3的全0张量 a=torch.zeros(2,3) print(a) # one torch b=torch.ones(2,3) # random torch c=torch.randn(2,3) # 从numpy中创建张量 import numpy as np numpy_array=np.array([1,2],[3,4]) tensor_from_numpy = torch.from_numpy(numpy_array) print(tensor_from_numpy) # 在指定的设备上创建张量 device=torch.device("cuda" if torch.cuda.is_available() else "cpu") d=torch.randn(2,3,device=device) print(d)