torch.tensor对应的是torch.DoubleTensor
torch.Tensor对应的是torch.FloatTensor
观察下 t的大小写
input = torch.Tensor(np.random.randn(4,3,256,256)).to(device)
torch.tensor对应的是torch.DoubleTensor
torch.Tensor对应的是torch.FloatTensor
观察下 t的大小写
input = torch.Tensor(np.random.randn(4,3,256,256)).to(device)