Path	Lines of Code
tf_agents/distributions/__init__.py	4
tf_agents/distributions/gumbel_softmax.py	36
tf_agents/distributions/masked.py	49
tf_agents/distributions/reparameterized_sampling.py	19
tf_agents/distributions/shifted_categorical.py	48
tf_agents/distributions/tanh_bijector_stable.py	27
tf_agents/distributions/utils.py	246