maxframe.tensor.tril#

maxframe.tensor.tril(m, k=0, gpu=None)[源代码]#

张量的下三角。

返回一个张量的副本，其中第 k 条对角线以上的元素被置为零。

参数:

  • m (array_like, shape (M, N)) -- 输入张量。

  • k (int, optional) -- 置零元素的对角线。k = 0`（默认值）表示主对角线，`k < 0 表示主对角线下方，k > 0 表示主对角线上方。

  • gpu (bool, optional) -- 如果为 True，则在 GPU 上分配张量，默认为 None

返回:

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

返回类型:

Tensor, shape (M, N)

参见

triu

类似的功能，但用于上三角

示例

>>> import maxframe.tensor as mt

>>> mt.tril([[1,2,3],[4,5,6],[7,8,9],[10,11,12]], -1).execute()
array([[ 0,  0,  0],
       [ 4,  0,  0],
       [ 7,  8,  0],
       [10, 11, 12]])