Remove usage of concrete.ndim
Merge pull request #40 from rhattersley/avoid_ndim

Remove usage of concrete.ndim