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, неявные нули останутся неизменными. Однако, если разреженный тензор содержит явные нули, они будут затронуты отображением!
| Args | |
|---|---|
op | Операция, которая должна быть применена к SparseTensor values. op обычно является поэлементной операцией (например, math_ops.add), но может быть использована любая операция, сохраняющая форму. |
*args | Аргументы для op. |
**kwargs | Параметры для op. |
| Returns | |
|---|---|
SparseTensor, чьи indices и dense_shape соответствуют indices и dense_shape всех входных SparseTensor. |
| Raises | |
|---|---|
ValueError | Если args не содержит SparseTensor или indices или dense_shape входных SparseTensor не равны. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/sparse/map_values