38 lines
1.7 KiB
ReStructuredText
38 lines
1.7 KiB
ReStructuredText
mindspore.mint.clamp
|
||
====================
|
||
|
||
.. py:function:: mindspore.mint.clamp(input, min=None, max=None) -> Tensor
|
||
|
||
将输入Tensor的值裁剪到指定的最小值和最大值之间。
|
||
|
||
限制 :math:`input` 的范围，其最小值为 `min` ，最大值为 `max` 。
|
||
|
||
.. math::
|
||
out_i= \left\{
|
||
\begin{array}{align}
|
||
max & \text{ if } input_i\ge max \\
|
||
input_i & \text{ if } min \lt input_i \lt max \\
|
||
min & \text{ if } input_i \le min \\
|
||
\end{array}\right.
|
||
|
||
.. note::
|
||
- `min` 和 `max` 不能同时为None；
|
||
- 当 `min` 为None，`max` 不为None时，Tensor中大于 `max` 的元素会变为 `max`；
|
||
- 当 `min` 不为None，`max` 为None时，Tensor中小于 `min` 的元素会变为 `min`；
|
||
- 当 `min` 大于 `max` 时，Tensor中所有元素的值会被置为 `max`；
|
||
- `input`，`min` 和 `max` 的数据类型需支持隐式类型转换，且不能为bool类型。
|
||
|
||
参数：
|
||
- **input** (Tensor) - `clamp` 的输入，类型为Tensor。支持任意维度的Tensor。
|
||
- **min** (Union(Tensor, float, int)，可选) - 指定最小值。默认值为 ``None`` 。
|
||
- **max** (Union(Tensor, float, int)，可选) - 指定最大值。默认值为 ``None`` 。
|
||
|
||
返回：
|
||
Tensor，表示裁剪后的Tensor。其shape和数据类型和 `input` 相同。
|
||
|
||
异常：
|
||
- **ValueError** - 如果 `min` 和 `max` 都为None。
|
||
- **TypeError** - 如果 `input` 的数据类型不是Tensor。
|
||
- **TypeError** - 如果 `min` 的数据类型不为None、Tensor、float或int。
|
||
- **TypeError** - 如果 `max` 的数据类型不为None、Tensor、float或int。
|