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