昇腾社区首页
中文
注册
开发者
下载

clip

函数功能

图像处理类,Tensor类的张量裁剪接口,用于将张量中的元素限制在给定的最小值和最大值之间(小于最小值的元素将会被替换为最小值、大于最大值的元素将会被替换为最大值)。

当前支持Atlas 推理系列产品Atlas 200I/500 A2 推理产品

  • 接口中的输入输出Tensor必须在Device侧。
  • 输入参数对应Tensor的形状(Shape)不超过4维。

函数原型

clip(inputTensor: Tensor, minVal: float, maxVal: float)

参数说明

参数名

输入/输出

说明

inputTensor

输入

Tensor类,支持dtype.float16、dtype.float32、dtype.uint8类型输入。

minVal

输入

float类型,张量中裁剪后的最小值,小于最小值的元素将会被替换为最小值。“minVal”需小于或等于“maxVal”。

maxVal

输入

float类型,张量中裁剪后的最大值,大于最大值的元素将会被替换为最大值。“maxVal”需大于或等于“minVal”。

返回参数说明

返回裁剪后的Tensor类数据。