生成随机数而不分配新的内存块

Question

tf.random.uniform没有out关键字。

在以下代码中，tensorflow 2.0是否会为100次分配2 x 2数组？

如果是这种情况，如何避免在GPU上进行不必要的分配？

谢谢。

测试

import tensorflow as tf
import numpy as np

for i in tf.range(tf.constant(100)):
    w = tf.random.uniform((2, 2), 0, 1, dtype=np.float32)
    # w is a tensorflow.python.framework.ops.EagerTensor
    print(w)