https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/networks/heteroscedastic_q_network/HeteroscedasticQNetwork?authuser=0
tf_agents.bandits.networks.heteroscedastic_q_network.HeteroscedasticQNetwork | TensorFlow Agents
Network Outputting Expected Value and Variance of Rewards.
tfagentsbanditsnetworksq