The class of `AffineGrid` has wrong docstring

[yiheng-wang-nv]

1. The values of parameters like rotate_params, shear_params and others are fixed, but the descriptions are like : generate the 1st rotation parameter from uniform ...

[Nic-Ma]

Sure, I will update the related docs and highlight reasonable input values.

Thanks.