flujo tensor:: operaciones:: SparseSegmentSumWithNumSegments
#include <math_ops.h>
Calcula la suma a lo largo de segmentos dispersos de un tensor.
Resumen
Como SparseSegmentSum
, pero permite identificadores faltantes en segment_ids
. Si falta una identificación, el tensor output
en esa posición se pondrá a cero.
Lea la sección sobre segmentación para obtener una explicación de los segmentos.
Por ejemplo:
c = tf.constant([[1,2,3,4], [-1,-2,-3,-4], [5,6,7,8]])
tf.sparse_segment_sum_with_num_segments(
c, tf.constant([0, 1]), tf.constant([0, 0]), num_segments=3)
# => [[0 0 0 0]
# [0 0 0 0]
# [0 0 0 0]]
tf.sparse_segment_sum_with_num_segments(c,
tf.constant([0, 1]),
tf.constant([0, 2],
num_segments=4))
# => [[ 1 2 3 4]
# [ 0 0 0 0]
# [-1 -2 -3 -4]
# [ 0 0 0 0]]
Argumentos:
- alcance: un objeto de alcance
- índices: un tensor 1-D. Tiene el mismo rango que
segment_ids
. - segment_ids: un tensor 1-D. Los valores deben ordenarse y pueden repetirse.
- num_segments: debe ser igual al número de ID de segmento distintos.
Devoluciones:
-
Output
: tiene la misma forma que los datos, excepto la dimensión 0 que tiene el tamañonum_segments
.
Constructores y destructores | |
---|---|
SparseSegmentSumWithNumSegments (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input indices, :: tensorflow::Input segment_ids, :: tensorflow::Input num_segments) |
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
SparseSegmentSumWithNumSegments
SparseSegmentSumWithNumSegments(
const ::tensorflow::Scope & scope,
::tensorflow::Input data,
::tensorflow::Input indices,
::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