return tf.random.stateless_uniform( tensorflow.python.framework.errors_impl.ResourceExhaustedError: {{function_node __wrapped__AddV2_device_/job:localhost/replica:0/task:0/device:GPU:0}} failed to allocate memory [Op:AddV2]

这个错误是 TensorFlow 运行时发出的，表示在执行 `tf.random.stateless_uniform` 操作时出现了内存不足的情况。具体来说，是因为显存不足导致 TensorFlow 无法分配内存，从而报告了这个错误。可能是因为模型太大或者数据量太大，导致显存不够。

解决方法可以尝试减小模型规模或者训练数据集的大小，或者使用更大的显存。另外，也可以尝试设置环境变量 `TF_GPU_ALLOCATOR=cuda_malloc_async` 来提高显存使用效率。如果这个错误频繁出现，建议检查一下显卡驱动是否最新、显存是否正常等硬件问题。