[functorch] Added im2col batch rule and enabled vmap for nn.functional.unfold op (pytorch/functorch#262)
Description:
- Added im2col batch rule and enabled vmap for nn.functional.unfold op
- Updated tests
Using EXISTING_BDIM macro to put bdim into 0 as im2col expects dim=0 to be batch dim
Related to pytorch/functorch#240