创建pytorch张量
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
下载地址
用户评论