in function random_affine, the h and w is wrong
in function random_affine, the h and w is wrong