[object Object]

[object Object][object Object]undefined
[object Object]
  • This API is used to pad the input tensor self based on the pad parameter. The padding value is value.

  • Formulas:

    • Shape deduction of the [object Object] tensor:

      Assumethattheshapeoftheinputselfisasfollows:[dim0in,dim1in,dim2in,dim3in]Assumethatpad=[dim3begin,dim3end,dim2begin,dim2end,dim1begin,dim1end,dim0begin,dim0end]\begin{aligned} Assume that the shape of the input self is as follows: \\ &[dim0_{in},dim1_{in}, dim2_{in}, dim3_{in}] Assume that \\ &pad = &[dim3_{begin},dim3_{end}, \\&&dim2_{begin},dim2_{end}, \\&&dim1_{begin},dim1_{end}, \\&&dim0_{begin},dim0_{end}] \end{aligned} Then,theshapeofoutisasfollows:[dim0out,dim1out,dim2out,dim3out]=[dim0begin+dim0in+dim0end,dim1begin+dim1in+dim1end,dim2begin+dim2in+dim2end,dim3begin+dim3in+dim3end]\begin{aligned} & Then, the shape of out is as follows: \\ &[dim0_{out}, dim1_{out}, dim2_{out}, dim3_{out}] = &[dim0_{begin}+dim0_{in}+dim0_{end}, \\&&dim1_{begin}+dim1_{in}+dim1_{end}, \\&&dim2_{begin}+dim2_{in}+dim2_{end}, \\&&dim3_{begin}+dim3_{in}+dim3_{end}] \end{aligned}
    • Example 1: (The length of the [object Object] array is twice the number of dimensions of [object Object].)

      selfShape=[1,1,1,1,1]pad={0,1,2,3,4,5,6,7,8,9}outputShape=[8+1+9,6+1+7,4+1+5,2+1+3,0+1+1]=[18,14,10,6,2]\begin{aligned} selfShape &= [1, 1, 1, 1, 1]\\ pad &= \lbrace 0, 1, 2, 3, 4, 5, 6, 7, 8, 9\rbrace \\ outputShape &= [8+1+9, 6+1+7, 4+1+5, 2+1+3, 0+1+1]\\ &= [18,14,10,6,2] \end{aligned}
    • Example 2: (The length of the [object Object] array is less than twice the number of dimensions of [object Object].)

      selfShape=[1,1,1,1,1]pad={0,1,2,3,4,5}outputShape=[0+1+0,0+1+0,4+1+5,2+1+3,0+1+1]=[1,1,10,6,2]\begin{aligned} selfShape &= [1, 1, 1, 1, 1]\\ pad &= \lbrace 0, 1, 2, 3, 4, 5\rbrace \\ outputShape &= [0+1+0, 0+1+0, 4+1+5, 2+1+3, 0+1+1]\\ &= [1,1,10,6,2] \end{aligned}
[object Object]

Each operator has calls. First, [object Object] is called to obtain the workspace size required for computation and the executor that contains the operator computation process. Then, [object Object] is called to perform computation.

[object Object]
[object Object]
[object Object]
  • Parameters

    [object Object]
    • For [object Object]Atlas training products[object Object] and [object Object]Atlas inference products[object Object]: The data type cannot be BFLOAT16, HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN or FLOAT8_E8M0.
    • For [object Object]Atlas A3 training products/Atlas A3 inference products[object Object] and [object Object]Atlas A2 training products/Atlas A2 inference products[object Object]: The data type cannot be HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN or FLOAT8_E8M0.
    • The data types of [object Object] and [object Object] must meet the data type deduction rules (see ).
    • If the data type of self is HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN or FLOAT8_E8M0, only the bit values of value can be all 0s.
  • 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]
[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]