摘要: Tensor类型 Tensor有不同的数据类型,每种类型分别有对应CPU和GPU版本(HalfTensor除外)。默认的Tensor是FloatTensor,可通过torch.set_default_tensor_type修改默认tensor类型(如果默认类型为GPU tensor,则所有操作都将在 阅读全文
posted @ 2021-04-23 15:26 EnjoyWorld 阅读(155) 评论(0) 推荐(0)