Redesign the tensor type

[kailuowang]

Should probably be just a high kinded type that takes in a dimension type param

[kailuowang]

or maybe just use Breeze types