module TensorStream::Train