fix import since move of *subtensor*
Add missing file from the refactoring.