Equivalent to:
meshes = []
for i in range(batch_size):
meshes.append(torch.meshgrid(tensor1[i], tensor2[i], ...))
And then stacking the zipped meshes along the batched dimension. Obviously this is really inefficient though.
Equivalent to:
meshes = []
for i in range(batch_size):
meshes.append(torch.meshgrid(tensor1[i], tensor2[i], ...))
And then stacking the zipped meshes along the batched dimension. Obviously this is really inefficient though.