Maybe I'm misunderstanding the nd array then. 

input is n dimensionnal, the cfa is gonna be the same thing + another dimensin of size 3.

For a 3D input, you get back a 4D output with the 4th dimension of length 3. That's what I meant in the docstring.