解决方法
原先代码
yv, xv = torch.meshgrid([torch.arange(hsize), torch.arange(wsize)], indexing="ij")
1
删除indexing="ij"即可
yv, xv = torch.meshgrid([torch.arange(hsize), torch.arange(wsize)])
解决方法
原先代码
yv, xv = torch.meshgrid([torch.arange(hsize), torch.arange(wsize)], indexing="ij")
1
删除indexing="ij"即可
yv, xv = torch.meshgrid([torch.arange(hsize), torch.arange(wsize)])