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)