Hi alban, Thanks for the suggestions! The ops in L are complex and cost much more time compared with other layers. I just use its implementation from other people[1] 5[2] 4 but none of them accept batch  I may try to reimplement it in the future.
Much appreciated!!