tensor2tensor
tensor2tensor copied to clipboard
question about inference output
Description
when i use tst-decoder, i want to get each timestep's probablity, such as 'a' [0.1, 0.1, 0.2.....]。i tried to add log parameter in decode_hparams, it didn't work。 i still want to control the decode_length to break the decode process before it meets end tag but not find the params。 does anyone know? ...
Environment information
OS: <your answer here>
$ pip freeze | grep tensor
# your output here
$ python -V
# your output here
For bugs: reproduction and error logs
# Steps to reproduce:
...
# Error logs:
...