def METHOD_NAME():
    bb = relax.BlockBuilder()
    g = relax.Var("g", R.Tensor((3, 2, 5), "float32"))
    x = relax.Var("x", R.Tensor((3, 4, 5), "float32"))
    indices = relax.Var("indices", R.Tensor((2,), "float32"))
    _check_inference(bb, relax.op.grad.take_backward(g, x, indices, axis=1), x.struct_info)