chainer.functions.batch_inv

chainer.functions.batch_inv(a)[source]

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