It only supports condensation based on the specific threshold then creates a new tensor.
We will merge it into the s2t_transformer.py finally.