affine_grid #18942

[so-dipe]

Close #18942

[fnhirwa]

Hey @so-dipe

Thanks for the PR 😊
Your changes looks good🙂 however, the tests are only passing for the Numpy and Jax backends would you look into this disparity?

Thanks

[so-dipe]

Hi...

My changes are in line with the affine_grid function in pytorch and they seem to be failing the same tests too. I think there's a general issue with the ivy implementation of some functions used there. I not sure which of the functions is causing the issue, and i didn't bother to check because I didn't think editing it would be in the cope of the issue.

[so-dipe]

Hi again @hirwa-nshuti
I suspect the problem is coming from here base_grid[:, :, :, 1] = ivy.expand_dims( ivy.linspace(-1, 1, H) * (H - 1) / H, axis=-1 ) numpy and jax handle the casting differently from the other frameworks

[so-dipe]

Hello @hirwa-nshuti, I have made some changes and I need you to take a look.

[fnhirwa]

Hey @so-dipe

Good progress with this item.
Now the failure is remaining with the paddle backend

[so-dipe]

it's a flaky test error... what am i going to do about that.🤷🏽‍♂️ @hirwa-nshuti

[so-dipe]

@hirwa-nshuti This is the issue. My code passes the test but fails it initially, there's nothing I can do about that. It's probably an error caused by another function or something.

[fnhirwa]

Looks great!
Going to merge it we will look into the detailed issue😊