分类：halfTensor

2021
07-12

pytorch--之halfTensor的使用详解

证明出错在dataloader里面在pytorch当中，float16和half是一样的数据结构，都是属于half操作，然后dataloader不能返回half值，所以在dataloader里面，要把float16改成float32即可返回补充：Pytorch中Tensor常用操作归纳对常用的一些Tensor的常用操作进行简单归纳，方便日后查询。后续有用到再补充。1、创建Tensorimporttorch#经典方式device=torch.device("cuda:0")x=torch.tensor([1,2],dtype=torch.float32,device=devi... 继续阅读 >