pytorch将其他类型转为tensor torch.as_tensor()、torch.from_numpy()

 

import torch
import numpy as np
a = np.array([1, 2, 3])
t = torch.as_tensor(a)
print(t)
t[0] = -1
a

 

将numpy转为tensor也可以使用t = torch.from_numpy(a)

posted @ 2022-03-18 18:53  Tomorrow1126  阅读(307)  评论(0编辑  收藏  举报