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)
wherem
is the number of matrices in the batch, andn
is the dimensionality of a square matrix.- Returns
Inverse of every matrix in the batch of matrices.
- Return type