max_val, max_idx = af.imax(img)

running this function seems to have wrong behavior and finds wrong max value for an image with pixels in range [0,1].

When running max_val, max_idx = af.imax(af.flat(img)) the correct max value is found.