ae31a3399b
* add batch_dim support * fix lint * add check for num of arguments for topi.take * fix gpu test cases * add check for batch_dims in take_grad