1. 首页
  2. 数据库
  3. 其它
  4. 创建pytorch张量

创建pytorch张量

上传者: 2020-12-22 21:52:11上传 PDF文件 36.71KB 热度 30次
1、numpy->tensor 注: tensor()参数为具体的数据, Tensor() 和 FloatTensor()参数可以为数据也可以为维度信息,但是一般不用他来放数据 a = torch.Tensor(2, 3) print(a) a = torch.FloatTensor(2, 3) print(a) a = np.array([2., 3]) b = torch.from_numpy(a) print(b) ################################### a = np.ones([2, 3]) b = torch.from_numpy(a) pr
下载地址
用户评论