如何在多个张量上并行执行torch.meshgrid



假设我们有一个大小为[60,9]的张量x和大小为[66,9]的张量y是否可以进行类似xx,yy = torch.meshgrid(x,y)的操作,使得xx和yy的大小为[60,9,9],并且xx[i,:,:], yy[i,:,:]基本上是torch.meshgrid(x[i],y[i])

内置的torch.meshgrid操作只接受1d张量,是否可以在不使用for循环的情况下进行上述操作(这是低效的,因为它不使用GPU的并行操作)?

我不相信你会得到任何东西,因为张量的初始化不是在GPU上完成的。因此,所提出的方法实际上是在xy上循环,或者使用map作为可迭代的:

grids = map(torch.meshgrid, zip(x,y))