Unfortunately this is not working for me… In tensorflow, the two tensors input and grid can have different number of dimensions
for eg:
input.get_shape() = [N, H, W, C] and
grid.get_shape() = [N, dim_0, ... , dim_n, 2]…
But torch.grid_sample() requires the input and grid to have the same number of dimensions.
Is there any solution to this.
My input is 4D and grid is 3D.
Thanks.