Я создаю слой, который использует свои входные тензоры для создания SparseTensor, т. Е. Входные тензоры являются соответствующими индексами и значениями SparseTensor.
С:
индексы: тензор 2-D int64 формы
И поскольку tf.cast(x, tf.int64) не дифференцируемо, я не уверен, достижимо ли это.
Другой вариант - найтиповорот на основе tf.round(), но SparseTensor не примет тензор другого типа в качестве индексов.
Есть ли способ привести тензор к целому числу и не иметь None для градиента?
Как я могу создать SparseTensor, используя выходные данные предыдущих слоев, которые являются тензорами с плавающей запятой?