2020
10-10
10-10
详解pytorch tensor和ndarray转换相关总结
在使用pytorch的时候,经常会涉及到两种数据格式tensor和ndarray之间的转换,这里总结一下两种格式的转换:1.tensorcpu和tensorgpu之间的转化:tensorcpu转为tensorgpu:tensor_gpu=tensor_cpu.cuda()>>>tensor_cpu=torch.ones((2,2))tensor([[1.,1.],[1.,1.]])>>>tensor_gpu=tensor_cpu.cuda()tensor([[1.,1.],[1.,1.]],device='cuda:0')tensorgpu转为tensorcpu:tensor_cpu=tensor_gpu.cuda()...
继续阅读 >