当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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


用法:

dask.array.tril(m, k=0)

数组的下三角形。

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

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

返回数组的副本,其中 k -th 对角线上方的元素为零。对于ndim 超过 2 的数组,tril 将应用于最后两个轴。

参数

m数组,形状(…,M,N)

输入数组。

k整数,可选

对角线高于其零元素。 k = 0(默认)是主对角线,k < 0在它下面,k > 0在上面。

返回

trilndarray,形状(...,M,N)

m 的下三角形,形状和数据类型与 m 相同。

例子

>>> np.tril([[1,2,3],[4,5,6],[7,8,9],[10,11,12]], -1)  
array([[ 0,  0,  0],
       [ 4,  0,  0],
       [ 7,  8,  0],
       [10, 11, 12]])
>>> np.tril(np.arange(3*4*5).reshape(3, 4, 5))  
array([[[ 0,  0,  0,  0,  0],
        [ 5,  6,  0,  0,  0],
        [10, 11, 12,  0,  0],
        [15, 16, 17, 18,  0]],
       [[20,  0,  0,  0,  0],
        [25, 26,  0,  0,  0],
        [30, 31, 32,  0,  0],
        [35, 36, 37, 38,  0]],
       [[40,  0,  0,  0,  0],
        [45, 46,  0,  0,  0],
        [50, 51, 52,  0,  0],
        [55, 56, 57, 58,  0]]])

相关用法


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