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_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()

你可能感兴趣的:(深度学习,pytorch)