- 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,)