Same comment here, it'd be nice if this could be public in someplace more tensor-oriented