Почему итерация по tf.Tensor не допускается в тензорном потоке?

Я использую @ tf.function для украшения функций для повышения скорости, но при попытке перебрать tf.range () выхожу с ошибкой. Эту ошибку можно воссоздать, используя следующий код

@tf.function
def loop(x, y):
    for i in tf.range(1,100,dtype=tf.int64):
        x.assign_add(y)
    return x

Моя версия тензорного потока - 1.15. Я не хочу заменять tf.range () диапазоном, так как, насколько мне известно, он развернет тело цикла. Есть ли способы решить эту проблему?

Информация об ОШИБКЕ ниже

OperatorNotAllowedInGraphError: iterating over `tf.Tensor` is not allowed: AutoGraph did not convert this function. Try decorating it directly with @tf.function