假设我们有一个大小为[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上完成的。因此,所提出的方法实际上是在x
和y
上循环,或者使用map
作为可迭代的:
grids = map(torch.meshgrid, zip(x,y))