title: can anyone give a tiny example to explain the params of tf.random.categorical?
body: <p>tensorflow's site gives this example</p>
  
  <pre><code>tf.random.categorical(tf.log([[10., 10.]]), 5)
  </code></pre>
  
  <p>produces a tensor that "has shape [1, 5], where each value is either 0 or 1 with equal probability"</p>
  
  <p>I have already known, the basic <a href="https://www.tensorflow.org/api_docs/python/tf/random/categorical" rel="noreferrer">demo</a>, the meaning of <code>tf.log([[10., 10.]])</code>.</p>
  
  <p>what I want to know is what does [batch_size, num_classes] do, can anyone give a tiny example to explain the params?</p>