- 博客(2)
- 收藏
- 关注
RSS订阅原创 pytorch中的tensor实现数据降维以及通道数转换
上例中torch.narrow(x,0,2,3)，因x为5X6的二维tensor，由两个维度[0,1]，0表示第一维,横向，1表示第二维，纵向；且第一维通道数为5，第二维通道数为6。第三个变量为所选维度中的第几个通道为起点。如果在深度学习中图像处理中常遇到思维tensor，即x.shape=[9，3，256，256],3代表彩色图片RGB的三个通道数，如想实现降维操作，可将x的第二维的通道数3变为1，再使用x=torch.squeeze(x,dim=1)实现降维，降维后shape变为[9,256,256]
2022-11-16 19:47:08
7487
1
原创 win10如何查看显卡CUDA的版本号以及显卡驱动程序版本号
点击键盘Win+R,输入cmd，在命令行继续输入nvidia-smi，即可查看
2022-06-09 20:34:58
1025
4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人