cupy.trace#
- cupy.trace(a, offset=0, axis1=0, axis2=1, dtype=None, out=None)[source]#
返回数组沿对角线的和。
它计算沿
axis1和axis2的对角线上的元素之和。- 参数:
a (cupy.ndarray) – 用于计算迹的数组。
offset (int) – 对角线的索引。零表示主对角线，正值表示上方对角线，负值表示下方对角线。
axis1 (int) – 用于计算迹的第一个轴。
axis2 (int) – 用于计算迹的第二个轴。
dtype – 输出数据类型指定符。
out (cupy.ndarray) – 输出数组。
- 返回值:
数组
a沿轴(axis1, axis2)的迹。- 返回值类型:
另请参阅