You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.nn.HSwish.rst 689 B

4 years ago
123456789101112131415161718192021222324252627
  1. mindspore.nn.HSwish
  2. ===================
  3. .. py:class:: mindspore.nn.HSwish
  4. Hard Swish激活函数。
  5. 对输入的每个元素计算Hard Swish。
  6. Hard Swish定义如下:
  7. .. math::
  8. \text{hswish}(x_{i}) = x_{i} * \frac{ReLU6(x_{i} + 3)}{6},
  9. 其中, :math:`x_i` 是输入的元素。
  10. **输入:**
  11. - **x** (Tensor) - 用于计算Hard Swish的Tensor。数据类型必须是float16或float32。shape为 :math:`(N,*)` ,其中 :math:`*` 表示任意的附加维度数。
  12. **输出:**
  13. Tensor,具有与 `x` 相同的数据类型和shape。
  14. **异常:**
  15. - **TypeError** - `x` 的数据类型既不是float16也不是float32。