cupy.inner

cupy.inner(a, b)

返回两个数组的内积。

它使用每个参数的最后一个轴来计算乘积和。

参数:

• a (cupy.ndarray) – 第一个参数。

• b (cupy.ndarray) – 第二个参数。

返回:

a 和 b 的内积。

返回类型:

cupy.ndarray

参见

numpy.inner()