chainer.functions.batch_inv¶

chainer.functions.batch_inv(a)[source]¶

Computes the inverse of a batch of square matrices.

Parameters: a (Variable or N-dimensional array) – 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