Calcula la suma a lo largo de segmentos dispersos de un tensor.
Lea [la sección sobre segmentación] (https://tensorflow.org/api_docs/python/tf/math#Segmentation) para obtener una explicación de los segmentos.
Como "SegmentSum", pero "ID_segmento" puede tener un rango menor que la primera dimensión de "datos", seleccionando un subconjunto de dimensión 0, especificado por "índices".
Por ejemplo:
c = tf.constant([[1,2,3,4], [-1,-2,-3,-4], [5,6,7,8]])
# Select two rows, one segment.
tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 0]))
# => [[0 0 0 0]]
# Select two rows, two segment.
tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 1]))
# => [[ 1 2 3 4]
# [-1 -2 -3 -4]]
# Select all rows, two segments.
tf.sparse_segment_sum(c, tf.constant([0, 1, 2]), tf.constant([0, 0, 1]))
# => [[0 0 0 0]
# [5 6 7 8]]
# Which is equivalent to:
tf.segment_sum(c, tf.constant([0, 0, 1]))
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
Salida <T> | asOutput () Devuelve el identificador simbólico del tensor. |
estática <T se extiende TNumber > SparseSegmentSum <T> | |
Salida <T> | salida () Tiene la misma forma que los datos, excepto por la dimensión 0 que tiene el tamaño `k`, el número de segmentos. |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Métodos públicos
pública de salida <T> asOutput ()
Devuelve el identificador simbólico del tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
public static SparseSegmentSum <T> Crear ( Alcance alcance, operando <T> de datos, operando <? extiende TNumber > índices, operando <? extiende TNumber > segmentIds)
Método de fábrica para crear una clase que envuelva una nueva operación SparseSegmentSum.
Parámetros
alcance | alcance actual |
---|---|
índices | Un tensor 1-D. Tiene el mismo rango que "ID_segmento". |
segmentoIds | Un tensor 1-D. Los valores deben ordenarse y pueden repetirse. |
Devoluciones
- una nueva instancia de SparseSegmentSum
pública de salida <T> salida ()
Tiene la misma forma que los datos, excepto por la dimensión 0 que tiene el tamaño `k`, el número de segmentos.