Python dask.array.trace用法及代码示例

用法: dask.array.trace(a, offset=0, axis1=0, axis2=1, dtype=None)

返回沿数组对角线的总和。

此文档字符串是从 numpy.trace 复制的。

可能存在与 Dask 版本的一些不一致之处。

如果a 是二维的，则返回沿其对角线与给定偏移量的总和，即所有 i 的元素 a[i,i+offset] 的总和。

如果a 有两个以上的维度，则由axis1 和axis2 指定的轴用于确定返回其轨迹的二维子数组。结果数组的形状与 a 的形状相同，但删除了 axis1 和 axis2。

a：array_like: 输入数组，从中获取对角线。
offset：整数，可选: 对角线与主对角线的偏移量。可以是正面的也可以是负面的。默认为 0。
axis1, axis2：整数，可选: 用作对角线的二维子阵列的第一和第二轴的轴。默认值是 a 的前两个轴。
dtype：dtype，可选: 确定返回数组和元素相加的累加器的数据类型。如果 dtype 的值为 None 并且 a 是精度小于默认整数精度的整数类型，则使用默认整数精度。否则，精度与 a 相同。
out：ndarray，可选(在 Dask 中不支持): 放置输出的数组。它的类型被保留，它必须具有正确的形状来保存输出。

sum_along_diagonals：ndarray: 如果a 是二维的，则返回沿对角线的和。如果a 具有更大的维度，则返回沿对角线的和数组。

>>> np.trace(np.eye(3))  
3.0
>>> a = np.arange(8).reshape((2,2,2))  
>>> np.trace(a)  
array([6, 8])

>>> a = np.arange(24).reshape((2,2,2,3))  
>>> np.trace(a).shape  
(2, 3)

相关用法

注：本文由纯净天空筛选整理自dask.org大神的英文原创作品 dask.array.trace。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。