def batch_matmul(a, b, transa=False, transb=False):
 return BatchMatMul(transa=transa, transb=transb)(a, b)

def batch_matmul ( a , b , transa = False , transb = False ) : return BatchMatMul ( transa = transa , transb = transb ) ( a , b )
computes the batch matrix multiplications of two sets of arrays .