refactor io module
pass tensor container and io