Not related to TF, but the upcoming version of Gorgonia has a solution for that - we basically developed a calculus for dimensionality of data. Gorgonia is deep learning in Go, and is well typed as far as I can tell.

Work on the coq program proving the calculus is taking longer than expected though.

If you're interested in contributing to Gorgonia, lmk.