2021
07-08
07-08
浅谈pytorch中stack和cat的及to_tensor的坑
初入计算机视觉遇到的一些坑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,...]...
继续阅读 >