[3.0] [future-work] Turn `TensorSerializer` into a context manager

[bchess]

from @Eta0 in #127 (review)

Turn TensorSerializer into a context manager to match with TensorDeserializer