Используйте словарь в tf.function input_signature в Tensorflow 2.0

Question

Я использую Tensorflow 2.0 и сталкиваюсь со следующей ситуацией:

@tf.function
def my_fn(items):
    .... #do stuff
    return

Если элементы являются указанием тензоров, например:

item1 = tf.zeros([1, 1])
item2 = tf.zeros(1)
items = {"item1": item1, "item2": item2}

Есть ли способ использования input_signature аргумент функции tf.function, чтобы я мог заставить tf2 избегать создания нескольких графиков, когда item1, например, tf.zeros([2,1])?