Pytorch中Tensor数据类型转换
Pytorch中Tensor数据类型转换:1)Tensor的int、float数据类型转换:在Tensor后加 .long(), .int(), .float(), .double()等即可2)Tensor与numpy数据类型转换:Tensor -> Numpy使用 data.numpy(),data为Tensor变量Numpy -> Tensor 使用 torch.from_nump
·
Pytorch中Tensor数据类型转换:
1)Tensor的int、float数据类型转换:
在Tensor后加 .long(), .int(), .float(), .double()等即可
2)Tensor与numpy数据类型转换:
Tensor -> Numpy 使用 data.numpy(),data为Tensor变量
Numpy -> Tensor 使用 torch.from_numpy(data),data为numpy变量
3)Tensor与Python数据类型转换:
Tensor -> 单个Python数据,使用data.item(),data为Tensor变量且只能为包含单个数据
Tensor -> Python list,使用data.tolist(),data为Tensor变量,返回shape相同的可嵌套的list
4)剥离出一个tensor参与计算,但不参与求导:
data.detach()
更多推荐
已为社区贡献2条内容
所有评论(0)