- 博客(2)
- 收藏
- 关注
RSS订阅原创 pytorch 张量与标量常见的一些操作
怎么把tensor数据类型转换为标量 a = torch.tensor([2.5]) a,a.item(),float(a),int(a) #结果是 (tensor([2.5000]), 2.5, 2.5, 2) tensor与标量基本的四则运算（加减乘除） 踩了一个大坑，tensor和列表也不能互相操作，所以折腾了很久发现必须改成numpy类型 函数 作用 add() Tensor中每个元素同加一个标量，或与另一个Tensor逐元素相加 mul() Tensor中每个元素同乘一...
2022-04-14 18:53:02
1401
原创 expected input to have 3 channels, but got 1 channels instead错误问题
从网上找的现成的代码，跑cifar10数据集没有问题，但是跑mnist数据集反而出了问题，是因为图片一个是彩色一个是灰度，在深度学习的时候，输入的channel不一样，找了很多资料，都是说让改成transforms.Lambda(lambda x: x.repeat(1,1,1))，还是不行，后来请教了同学，改成了transforms.Lambda(lambda x: x.repeat(3,1,1))，就OK了 ...
2022-04-05 00:23:20
4891
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人