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

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐