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

时间:2019-06-12 19:42:33

标签: python tensorflow tensorflow2.0

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)

0 个答案:

没有答案