tril

paddle.tril ( input, diagonal=0, name=None ) [源代码]

返回输入矩阵 input 的下三角部分，其余部分被设为0。 矩形的下三角部分被定义为对角线上和下方的元素。

参数

  • input (Tensor) : 输入Tensor input，数据类型支持 float32, float64, int32, int64 。

  • diagonal (int，可选) : 指定的对角线，默认值为0。如果diagonal = 0，表示主对角线; 如果diagonal是正数，表示主对角线之上的对角线; 如果diagonal是负数，表示主对角线之下的对角线。

  • name (str，可选)- 具体用法请参见 Name ，一般无需设置，默认值为None。

返回

Tensor, 数据类型与输入 input 数据类型一致。

代码示例

  1. import numpy as np
  2. import paddle
  4. data = np.arange(1, 13, dtype="int64").reshape(3,-1)
  5. # array([[ 1,  2,  3,  4],
  6. #        [ 5,  6,  7,  8],
  7. #        [ 9, 10, 11, 12]])
  10. x = paddle.to_tensor(data)
  12. tril1 = paddle.tensor.tril(x)
  13. # array([[ 1,  0,  0,  0],
  14. #        [ 5,  6,  0,  0],
  15. #        [ 9, 10, 11,  0]])
  17. # example 2, positive diagonal value
  18. tril2 = paddle.tensor.tril(x, diagonal=2)
  19. # array([[ 1,  2,  3,  0],
  20. #        [ 5,  6,  7,  8],
  21. #        [ 9, 10, 11, 12]])
  23. # example 3, negative diagonal value
  24. tril3 = paddle.tensor.tril(x, diagonal=-1)
  25. # array([[ 0,  0,  0,  0],
  26. #        [ 5,  0,  0,  0],
  27. #        [ 9, 10,  0,  0]])