Tensorflow text_generation

Question

我正在处理代码

https://www.tensorflow.org/tutorials/sequences/text_generation

当我到达该行时，会产生以下错误。

 sampled_indices = tf.random.categorical(example_batch_predictions[0], num_samples=1)

 sampled_indices = tf.squeeze(sampled_indices,axis=-1).numpy()

错误

AttributeError

Traceback (most recent call last)

在

----> 1 sampled_indices = tf.random.categorical(example_batch_predictions[0], num_samples=1)

2 sampled_indices = tf.squeeze(sampled_indices,axis=-1).numpy()

AttributeError: module 'tensorflow._api.v1.random' has no attribute 'categorical'

系统信息 - TensorFlow 版本：Ubuntu 上的 1.12 Jupyter NoteBooks

可能缺少什么的任何指示？我想知道是否缺少导入？