cupy.linalg.tensorsolve#
- cupy.linalg.tensorsolve(a, b, axes=None)[源代码]#
求解张量方程 ax = b。
假设 b 等价于 cupy.tensordot(a, x)。此函数从 a 和 b 计算张量 x。
- 参数：
a (cupy.ndarray) – 形状长度 >= 1 的张量
b (cupy.ndarray) – 形状长度 >= 1 的张量
axes (整数元组) – 在求逆前需要重新排列到右侧的 a 的轴。
- 返回值：
形状为 Q 的张量，使得 b.shape + Q == a.shape。
- 返回类型：
警告
此函数调用一个或多个 cuSOLVER 例程，如果输入条件不满足，可能会产生无效结果。要检测这些无效结果，您可以在
cupyx.errstate()或cupyx.seterr()中将 linalg 配置设置为非 ignore 的值。