x, y = torch.meshgrid(torch.arange(width, device=device), torch.arange(height, device=device))生成的结果没有grad_fn属性该怎么改进
时间: 2023-06-07 10:09:18 浏览: 80
Pytorch训练过程中改变模型参数 requires_grad 属性
5星 · 资源好评率100%
可以使用 torch.meshgrid 中的参数 indexing='ij' 来解决没有 grad_fn 属性的问题。具体实现代码如下:
x, y = torch.meshgrid(torch.arange(width, device=device), torch.arange(height, device=device), indexing='ij')
阅读全文