Динамическая ссылка на разреженный тензор в списке - PullRequest
0 голосов
/ 07 августа 2020

Я пытаюсь динамически ссылаться на разреженный тензор из списка, учитывая одномерный тензор индексов.


# list of sparse tensors (different dimensions):
sparse_list = [sp1, sp2, sp3]

# tensor of indices to reference sparse tensors:
select_sparse_idx = tf.convert_to_tensor([1, 1, 0])

# within a tf.while_loop, where t is the iterable:
sparse_tensor = sparse_list[select_sparse_idx[t]]

# Error raised:
TypeError: list indices must be integers or slices, not Tensor

Если бы тензоры были плотными, я бы упаковал их в трехмерную структуру данных, где внешнее измерение было измерением выбора. Здесь я не могу упаковать разреженные тензоры вместе, потому что размеры базовых структур данных не обязательно будут одинаковыми.

...