Skip to content

Latest commit

 

History

History
110 lines (98 loc) · 3.81 KB

File metadata and controls

110 lines (98 loc) · 3.81 KB

miscellaneous

目前已完成的API: Clip (Tensor, Scalar, Scalar), Clip (Tensor, Scalar, Tensor), Clip (Tensor, Tensor, Scalar), Clip (Tensor, Tensor, Tensor), Fabs, Fmax, Fmin, Maximum, Minimum, Nan_to_num, Square

1. Clip(Tensor, Scalar, Scalar)

  • 参数:
    a:NPUArray,输入数组
    a_min:py::object,最小值(标量)
    a_max:py::object,最大值(标量)
  • 返回类型:
    NPUArray
  • 功能:
    将输入数组裁剪到区间 [a_min, a_max]。

2. Clip(Tensor, Scalar, Tensor)

  • 参数:
    a:NPUArray,输入数组
    a_min:py::object,最小值(标量)
    a_max:NPUArray,最大值
  • 返回类型:
    NPUArray
  • 功能:
    将输入数组逐元素裁剪,最小值为标量,最大值来自数组。

3. Clip(Tensor, Tensor, Scalar)

  • 参数:
    a:NPUArray,输入数组
    a_min:NPUArray,最小值
    a_max:py::object,最大值(标量)
  • 返回类型:
    NPUArray
  • 功能:
    将输入数组逐元素裁剪,最小值为数组,最大值为标量。

4. Clip(Tensor, Tensor, Tensor)

  • 参数:
    a:NPUArray,输入数组
    a_min:NPUArray,最小值
    a_max:NPUArray,最大值
  • 返回类型:
    NPUArray
  • 功能:
    将输入数组逐元素裁剪到对应区间。

5. Fabs

  • 参数:
    x:NPUArray,输入数组
  • 返回类型:
    NPUArray
  • 功能:
    计算逐元素绝对值(等同 Absolute)。

6. Fmax

  • 参数:
    x1:NPUArray,输入数组
    x2:NPUArray,输入数组
    dtype:py::dtype,结果类型
  • 返回类型:
    NPUArray
  • 功能:
    逐元素取较大值(忽略 NaN 行为同 maximum)。

7. Fmin

  • 参数:
    x1:NPUArray,输入数组
    x2:NPUArray,输入数组
    dtype:py::dtype,结果类型
  • 返回类型:
    NPUArray
  • 功能:
    逐元素取较小值(忽略 NaN 行为同 minimum)。

8. Maximum

  • 参数:
    x1:NPUArray,输入数组
    x2:NPUArray,输入数组
    dtype:py::dtype,结果类型
  • 返回类型:
    NPUArray
  • 功能:
    逐元素返回最大值。

9. Minimum

  • 参数:
    x1:NPUArray,输入数组
    x2:NPUArray,输入数组
    dtype:py::dtype,结果类型
  • 返回类型:
    NPUArray
  • 功能:
    逐元素返回最小值。

10. Nan_to_num

  • 参数:
    x:NPUArray,输入数组
    nan:double,用于替换 NaN 的值
    posinf:py::object,替换正无穷
    neginf:py::object,替换负无穷
  • 返回类型:
    NPUArray
  • 功能:
    替换数组中的 NaN、+inf 和 -inf。

11. Square

  • 参数:
    x:NPUArray,输入数组
  • 返回类型:
    NPUArray
  • 功能:
    逐元素计算平方。