分类：全部列表
在看程序的时候看到了x[…,0]的语句不是很理解，后来自己做实验略微了解，以此记录方便自己查看。b=torch.Tensor([[[[10,2],[4,5],[7,8]],[[1,2],[4,5],[7,8]]]])print(b.size())(1,2,3,2)print(b[…,0])tensor([[[10.,4.,7.],[1.,4.,7.]]])print(b[…,0].size())(1,2,3)print(b[…,2])Traceback(mostrecentcalllast):File“”,line1,inIndexError:index2isoutofboundsfordimension3withsize2print...
继续阅读 >
初入计算机视觉遇到的一些坑1.pytorch中转tensorx=np.random.randint(10,100,(10,10,10))x=TF.to_tensor(x)print(x)这个函数会对输入数据进行自动归一化，比如有时候我们需要将0-255的图片转为numpy类型的数据，则会自动转为0-1之间2.stack和cat之间的差别stackx=torch.randn((1,2,3))y=torch.randn((1,2,3))z=torch.stack((x,y))#默认dim=0print(z.shape)#torch.Size([2,1,2,3])所以stack的之后的数据也就很好理解了，z[0,...]...
继续阅读 >
需求：对于一个pythonlist或者numpy数组，我需要找到这个list中最大的K个数及其对应的下标。解决方式：1.可以构造字典通过排序解决，不过代码量较多。2.使用heapq库，可以直接获取最大值的下标和数值。importheapqa=[4,2,6,1,9,9]#获取下标，输出为[4,5,2]heapq.nlargest(3,range(len(a)),a.__getitem__)#获取数值，输出为[9,9,6]heapq.nlargest(3,a)如果要取最小的数，使用nsmallest即可补充：Python利用中...
继续阅读 >
