首页 > 试题广场 >

pytorch中cuda()作用,两个Tensor,一个加了

[问答题]

请你思考一下,pytorch中cuda()作用,两个Tensor,一个加了cuda(),一个没加,相加后会发生什么情况?

类型不匹配,gpu版本是torch.cuda.FloatTensor,cpu版本是torch.FloatTensor
发表于 2019-06-19 21:39:48 回复(0)
cuda放到了gpu上,CPU和GPU的tensor不能直接想加,会报错
发表于 2019-03-18 11:26:46 回复(1)
类型不匹配
发表于 2019-03-15 08:57:17 回复(1)
cuda()用于将变量传输到GPU上;相加之后会报错:变量类型不匹配,无法运算。
发表于 2020-02-25 15:22:51 回复(1)
类型不匹配
发表于 2019-04-16 20:57:30 回复(0)