cupy.linalg.tensorinv#

cupy.linalg.tensorinv(a, ind=2)[源代码][源代码]#

计算张量的逆。

此函数从张量 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] 等价的张量的逆。

返回类型:

cupy.ndarray

警告

此函数调用一个或多个 cuSOLVER 例程，如果输入条件不满足，可能会产生无效结果。要检测这些无效结果，您可以在 cupyx.errstate() 或 cupyx.seterr() 中将 linalg 配置设置为非 ignore 的值。

参见

numpy.linalg.tensorinv()