当前位置:   article > 正文

Pytorch中tensor的数据类型显示和转换方法_查看tensor数据类型

查看tensor数据类型

1.显示pytorch中tensor的数据类型:

  1. import torch
  2. x=torch.Tensor([1,2])
  3. print('x: ',x)
  4. print('type(x): ',x.size() ) # 查看tensor的维度是什么样,查看tensor的shape
  5. print('x.dtype: ',x.dtype) # 这个才是查看tensor中数据的具体类型是什么

pytorch: 获取 tensor 维度的方法:shape 和 size()_Caesar6666的博客-CSDN博客_查看tensor的shape

2. 如何转换tensor中数据的数据类型

x = torch.tensor(x, dtype=torch.float64)

数据类型转换成:float64类型。

会提示一个warning : UserWarning: To copy construct from a tensor, it is recommended to use sourceTensor.clone().detach() or sourceTensor.clone().detach().requires_grad_(True), rather than torch.tensor(sourceTensor). 

但是不用管,,,

3,一些疑惑待解决:

PS:但有的教程给出了还要将x = variable(x) 的方式,不是很明白。但是链接放在这:

Pytorch的Variable详解_Miss_zhuo_的博客-CSDN博客_pytorch variable

【pytorch错误】:Pytorch RuntimeError: “host_softmax” not implemented for 'torch.cuda.LongTensor'_suyunzzz的博客-CSDN博客

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/272350?site
推荐阅读
相关标签
  

闽ICP备14008679号