可变长度输入的内存泄漏

时间:2019-10-01 22:35:28

标签: tensorflow2.0

TensorFlow 2.0中的以下代码(10月1日每晚)是否会导致内存泄漏?

import tensorflow as tf

layer = tf.keras.layers.LSTM(units=522)

for _ in range(502):
    ts = tf.random.uniform(shape=(), minval=14, maxval=357, dtype=tf.int32)
    inputs = tf.random.uniform([65, ts, 513])
    _ = layer(inputs)

如果是,建议的缓解方法是什么?

0 个答案:

没有答案