cupy.linalg.tensorinv#
- cupy.linalg.tensorinv(a, ind=2)[source]#
计算张量的逆。
此函数根据张量
a计算张量a_inv，使得tensordot(a_inv, a, ind) == I，其中I表示单位张量。- 参数:
a (cupy.ndarray) – 输入张量，满足
prod(a.shape[:ind]) == prod(a.shape[ind:])。ind (int) – 在
tensordot函数的axes选项中使用的正数。
- 返回值:
张量的逆，其形状等同于
a.shape[ind:] + a.shape[:ind]。- 返回类型:
警告
此函数调用一个或多个 cuSOLVER 例程，如果输入条件不满足，可能会产生无效结果。为了检测这些无效结果，您可以在
cupyx.errstate()或cupyx.seterr()中将 linalg 配置设置为非 ignore 的值。