flujo tensor:: operaciones:: Suma de segmento sin clasificar
#include <math_ops.h>
Calcula la suma a lo largo de segmentos de un tensor.
Resumen
Lea la sección sobre segmentación para obtener una explicación de los segmentos.
Calcula un tensor tal que \(output[i] = {j...} data[j...]\) donde la suma es sobre tuplas j...
tal que segment_ids[j...] == i
. A diferencia de SegmentSum
, no es necesario ordenar segment_ids
y no es necesario cubrir todos los valores en el rango completo de valores válidos.
Si la suma está vacía para un ID de segmento determinado i
, output[i] = 0
. Si el ID de segmento dado i
es negativo, el valor se descarta y no se agregará a la suma del segmento.
num_segments
debe ser igual al número de ID de segmento distintos.
c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]]) tf.unsorted_segment_sum(c, tf.constant([0, 1, 0]), num_segments=2) # ==> [[ 5, 5, 5, 5], # [5, 6, 7, 8]]
Argumentos:
- alcance: un objeto de alcance
- segment_ids: un tensor cuya forma es un prefijo de
data.shape
.
Devoluciones:
-
Output
: tiene la misma forma que los datos, excepto las primeras dimensionessegment_ids.rank
, que se reemplazan con una única dimensión que tiene el tamañonum_segments
.
Constructores y destructores | |
---|---|
UnsortedSegmentSum (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input segment_ids, :: tensorflow::Input num_segments) |
Atributos públicos | |
---|---|
operation | |
output |
Funciones públicas | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Atributos públicos
operación
Operation operation
producción
::tensorflow::Output output
Funciones públicas
Suma de segmento sin clasificar
UnsortedSegmentSum( const ::tensorflow::Scope & scope, ::tensorflow::Input data, ::tensorflow::Input segment_ids, ::tensorflow::Input num_segments )
nodo
::tensorflow::Node * node() const
operador::tensorflow::Entrada
operator::tensorflow::Input() const
operador::tensorflow::Salida
operator::tensorflow::Output() const