Description: Sets the elements of the input tensor self (last two dimensions) to 0 in the lower triangular part below the main diagonal. The diagonal parameter can be set to a positive or negative value. The default value is 0. A positive value indicates diagonals above the main diagonal. A negative value indicates diagonals below the main diagonal.
Formula: In the following formula, i indicates the sequence number of the element in the second-to-last dimension (i is the row index), j indicates the sequence number of the element in the last dimension (j is the column index), and d indicates diagonal. In the two-dimensional coordinate diagram corresponding to (i, j), i+d==j indicates that the element is on the diagonal.
Example:
, The result of triu(self, diagonal=0) is as follows: ; Adjust the value of diagonal. The result of triu(self, diagonal=1) is as follows: ; Adjust the value of diagonal to -1. The result of triu(self, diagonal=-1) is as follows: .
- aclnnTriu and aclnnInplaceTriu implement the same function in different ways. Select a proper operator based on your requirements.
- aclnnTriu: An output tensor object needs to be created to store the computation result.
- aclnnInplaceTriu: No output tensor object needs to be created, and the computation result is stored in the memory of the input tensor.
- Each operator has calls. First,
[object Object]or[object Object]is called to obtain input parameters and calculate the required workspace size based on the computation process. Then,[object Object]or[object Object]is called to perform computation.
Parameters
[object Object]- Ascend 950PR/Ascend 950DT: Only COMPLEX32 and COMPLEX64 are supported.
- [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type cannot be BFLOAT16.
Return Value
[object Object]: status code. For details, see .The first-phase API implements input parameter validation. The following error codes may be returned.
[object Object]
Parameters
[object Object]- Ascend 950PR/Ascend 950DT: Only COMPLEX32 and COMPLEX64 are supported.
- [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type cannot be BFLOAT16.
Return Value
[object Object]: status code. For details, see .The first-phase API implements input parameter validation. The following error codes may be returned.
[object Object]
- Deterministic computation:
[object Object]defaults to a deterministic implementation.
The following example is for reference only. For details, see .