分类：0]

2021
07-08

pytorch中[..., 0]的用法说明

在看程序的时候看到了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... 继续阅读 >

2021
06-23

Python中X[:,0]和X[:,1]的用法

X[:,0]是numpy中数组的一种写法，表示对一个二维数组，取该二维数组第一维中的所有数据，第二维中取第0个数据，直观来说，X[:,0]就是取所有行的第0个数据,X[:,1]就是取所有行的第1个数据。举例说明：importnumpyasnpX=np.array([[0,1],[2,3],[4,5],[6,7],[8,9],[10,11],[12,13],[14,15],[16,17],[18,19]])printX[:,0]X[:,0]输出结果是：importnumpyasnpX=np.array([[0,1],[2,3],[4,5],[6,7],[8,9],[10,11],[12,13],... 继续阅读 >