tf.sparse.map_values

Применяет op к тензору .values одного или нескольких SparseTensor.

tf.sparse.map_values(
    op, *args, **kwargs
)

Заменяет любые SparseTensor в args или kwargs на соответствующий им тензор values (который содержит значения, отличные от значений по умолчанию для SparseTensor), а затем вызывает op. Возвращает SparseTensor, построенный из входных SparseTensor indices, dense_shape, и значения, возвращённого op.

Если входные аргументы содержат несколько SparseTensor , то они должны иметь одинаковые размеры indices и плотные формы.

Примеры:

s = tf.sparse.from_dense([[1, 2, 0],
                          [0, 4, 0],
                          [1, 0, 0]])
tf.sparse.to_dense(tf.sparse.map_values(tf.ones_like, s)).numpy()
array([[1, 1, 0],
       [0, 1, 0],
       [1, 0, 0]], dtype=int32)

tf.sparse.to_dense(tf.sparse.map_values(tf.multiply, s, s)).numpy()
array([[ 1,  4,  0],
       [ 0, 16,  0],
       [ 1,  0,  0]], dtype=int32)

tf.sparse.to_dense(tf.sparse.map_values(tf.add, s, 5)).numpy()
array([[6, 7, 0],
       [0, 9, 0],
       [6, 0, 0]], dtype=int32)

Примечание: несмотря на то, что tf.add(0, 5) != 0, неявные нули останутся неизменными. Однако, если разреженный тензор содержит явные нули, они будут затронуты отображением!