cupy.trace#

cupy.trace(a, offset=0, axis1=0, axis2=1, dtype=None, out=None)[源代码][源代码]#

返回数组对角线上的元素之和。

它计算沿 axis1 和 axis2 对角线上的和。

参数:

  • a (cupy.ndarray) – 要取迹的数组。

  • offset (int) – 对角线索引。零表示主对角线，正值表示上对角线，负值表示下对角线。

  • axis1 (int) – 沿着第一个轴进行迹运算。

  • axis2 (int) – 取迹的第二个轴。

  • dtype – 输出数据类型的指定符。

  • out (cupy.ndarray) – 输出数组。

返回:

沿轴 (axis1, axis2) 的 a 的轨迹。

返回类型:

cupy.ndarray

参见

numpy.trace()