cupy.linalg.tensorsolve#

cupy.linalg.tensorsolve(a, b, axes=None)[源代码][源代码]#

求解由 ax = b 表示的张量方程。

假设 b 等同于 cupy.tensordot(a, x)。此函数根据 a 和 b 计算张量 x。

参数:

  • a (cupy.ndarray) – 张量的 len(shape) >= 1

  • b (cupy.ndarray) – 张量的 len(shape) >= 1

  • axes (tuple of ints) – 在 a 中的轴，在反转之前重新排序到右侧。

返回:

形状为 Q 的张量，使得 b.shape + Q == a.shape。

返回类型:

cupy.ndarray

警告

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

参见

numpy.linalg.tensorsolve()