The output shape is described in the docs 12 and explains that dim1 will hold the window values.
In your use case
print(output[0, :, 0])
> tensor([0., 0., 0., 0., 0., 1., 0., 5., 6.])

will print the expected result.