Function torch::linalg::tensorinv¶
Defined in File linalg.h
Function Documentation¶
-
inline Tensor torch::linalg::tensorinv(const Tensor &self, int64_t ind)¶
Computes the inverse of a tensor.
See https://pytorch.org/docs/main/linalg.html#torch.linalg.tensorinv
Example:
auto a = torch::eye(4*6).reshape({4, 6, 8, 3}); int64_t ind = 2; auto ainv = torch::linalg::tensorinv(a, ind);