chainer.functions.batch_inv

chainer.functions.batch_inv(a)

Computes the inverse of a batch of square matrices.

Parameters:a (Variable) – Input array to compute the inverse for. Shape of the array should be (m, n, n) where m is the number of matrices in the batch, and n is the dimensionality of a square matrix. Returns:Inverse of every matrix in the batch of matrices. Return type:Variable