torch.nn.functional.ema

torch.nn.functional.ema 函数实现了指数移动平均的计算，其用法为：

torch.nn.functional.ema(tensor, decay, axis=None, offset=None, dtype=None, name=None)

参数说明：

- `tensor`：输入张量。
- `decay`：衰减系数，取值范围为(0, 1]，越接近1则计算结果越接近输入的张量。
- `axis`：在哪个维度上计算指数移动平均，默认为 None，表示在整个张量上计算。
- `offset`：标量偏移量，用于在起始时刻计算指数移动平均值，默认为 None。
- `dtype`：输出张量的数据类型，默认为 None，表示与输入张量一致。
- `name`：操作的名称。

该函数的作用是对输入张量进行指数移动平均的计算，其公式为：

EMA_t = decay * EMA_{t-1} + (1-decay) * x_t

其中，`t` 表示当前时刻，`EMA_t` 表示当前时刻的指数移动平均值，`x_t` 表示当前时刻的输入张量。通过不断迭代计算，可以得到每个时刻的指数移动平均值。