pcth=np.repeat(np.arange(height).reshape(height,1),[width],axis=1)怎么用torch模块中的函数表示
时间: 2023-05-14 10:06:13 浏览: 204
浅谈python中np.array的shape( ,)与( ,1)的区别
5星 · 资源好评率100%
您可以使用torch.from_numpy函数将np.repeat(np.arange(height).reshape(height,1),[width],axis=1)转换为torch张量,如下所示:
pcth = torch.from_numpy(np.repeat(np.arange(height).reshape(height,1),[width],axis=1))
阅读全文