[object Object]

[object Object][object Object]undefined
[object Object]
  • Description: Sets the elements of the input tensor [object Object] (last two dimensions) to [object Object] in the upper triangular part above the main diagonal. The diagonal parameter can be set to a positive or negative value. The default value is [object Object]. A positive value indicates diagonals above the main diagonal. A negative value indicates diagonals below the main diagonal.

  • Formula: In the following formula, [object Object] indicates the sequence number of the element in the second-to-last dimension ([object Object] is the row index), [object Object] indicates the sequence number of the element in the last dimension ([object Object] is the column index), and [object Object] indicates diagonal. In the two-dimensional coordinate diagram corresponding to (i, j), i+d==j indicates that the element is on the diagonal.

    Diagonalandlowertriangularpart,thatis,i+dj.Retaintheoriginalvalue:outi,j=selfi,j.Uppertriangularpart,thatis,i+d<j.Elementsaresetto0(excludingthediagonal):outi,j=0.Diagonal and lower triangular part, that is, i+d ≥ j. Retain the original value: out_{i, j} = self_{i, j}\\. Upper triangular part, that is, i+d < j. Elements are set to `0` (excluding the diagonal): out_{i, j} = 0.
  • Example:

    self=[[963][123][341]]self = \begin{bmatrix} [9&6&3] \\ [1&2&3] \\ [3&4&1] \end{bmatrix}, The result of triu(self, diagonal=0) is as follows:

    [[900][120][341]]\begin{bmatrix} [9&0&0] \\ [1&2&0] \\ [3&4&1] \end{bmatrix};

    Adjust the value of [object Object]. The result of triu(self, diagonal=1) is as follows:

    [[960][123][341]]\begin{bmatrix} [9&6&0] \\ [1&2&3] \\ [3&4&1] \end{bmatrix};

    Adjust the value of [object Object] to [object Object]. The result of triu(self, diagonal=-1) is as follows:

    [[000][100][340]]\begin{bmatrix} [0&0&0] \\ [1&0&0] \\ [3&4&0] \end{bmatrix}.

[object Object]
  • [object Object] and [object Object] implement the same function in different ways. Select a proper operator based on your requirements.
    • [object Object]: An output tensor object needs to be created to store the computation result.
    • [object Object]: 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.
[object Object]
[object Object]
[object Object]
[object Object]
[object Object]
  • Parameters

    [object Object]
  • Returns

    [object Object]: status code. For details, see .

    The first-phase API performs input parameter validation. The following errors may be returned:

    [object Object]
[object Object]
  • Parameters

    [object Object]
  • Returns

    [object Object]: status code. For details, see .

[object Object]
  • Parameters

    [object Object]
  • Returns

    [object Object]: status code. For details, see .

    The first-phase API performs input parameter validation. The following errors may be returned:

    [object Object]
[object Object]
  • Parameters

    [object Object]
  • Returns

    [object Object]: status code. For details, see .

[object Object]
  • Deterministic computation:
    • [object Object] defaults to a deterministic implementation.
[object Object]

The following example is for reference only. For details, see .

[object Object]