bartorch.moving_average

bartorch.moving_average(input, axis, length)

Mean over each window of length consecutive elements along axis (-a, -l).

Only whole windows are taken: the axis shrinks from n to n - length + 1 and out[i] is the mean of input[i:i + length].