cupy.linalg.tensorsolve¶
-
cupy.linalg.
tensorsolve
(a, b, axes=None)[source]¶ Solves tensor equations denoted by
ax = b
.Suppose that
b
is equivalent tocupy.tensordot(a, x)
. This function computes tensorx
froma
andb
.Parameters: - a (cupy.ndarray) – The tensor with
len(shape) >= 1
- b (cupy.ndarray) – The tensor with
len(shape) >= 1
- axes (tuple of ints) – Axes in
a
to reorder to the right before inversion.
Returns: The tensor with shape
Q
such thatb.shape + Q == a.shape
.Return type: See also
- a (cupy.ndarray) – The tensor with