class mindspore_rl.utils.OUNoise(stddev, damping, action_shape)[源代码]

在action上加入Ornstein-Uhlenbeck (OU)噪声。

参数:
  • stddev (float) - Ornstein-Uhlenbeck (OU) 噪声标准差。

  • damping (float) - Ornstein-Uhlenbeck (OU) 噪声阻尼。

  • action_shape (tuple) - 动作的维度。

输入:
  • actions (Tensor) - 添加OU噪声之前的动作。

输出:
  • actions (Tensor) - 添加OU噪声之后的动作。

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> from mindspore_rl.utils import OUNoise
>>> action_shape = (6,)
>>> actions = Tensor(np.ones(action_shape))
>>> net = OUNoise(stddev=0.2, damping=0.15, action_shape=action_shape)
>>> actions = net(actions)
>>> print(actions.shape)
(6,)