mindchemistry.e3.nn.soft_unit_step

mindchemistry.e3.nn.soft_unit_step(x)

单位阶跃函数的平滑版本。

\[x \mapsto \theta(x) e^{-1/x}\]

参数：

• x (Tensor) - 输入张量。

返回：

张量，单位阶跃函数的输出。

支持平台：

Ascend

样例：

>>> from mindchemistry.e3.nn import soft_unit_step
>>> from mindspore import ops, set_context, Tensor
>>> x = Tensor(ops.linspace(-1.0, 10.0, 1000))
>>> outputs = soft_unit_step(x)
>>> print(outputs.shape)
(1000,)