- 博客(2)
- 收藏
- 关注
原创 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 7799 1
原创 win10如何查看显卡CUDA的版本号以及显卡驱动程序版本号
点击键盘Win+R,输入cmd,在命令行继续输入nvidia-smi,即可查看
2022-06-09 20:34:58 1080 4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人