tensorflow :: operaciones :: UnsortedSegmentProd
#include <math_ops.h>
Calcula el producto a lo largo de los segmentos de un tensor.
Resumen
Lea la sección sobre segmentación para obtener una explicación de los segmentos.
Este operador es similar al operador de suma de segmentos sin clasificar encontrado (aquí) . En lugar de calcular la suma de los segmentos, calcula el producto de todas las entradas que pertenecen a un segmento de manera que:
\(output_i = {j...} data[j...]\) donde el producto está por encima de tuplas j...
tal que segment_ids[j...] == i
.
Por ejemplo:
c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]]) tf.unsorted_segment_prod(c, tf.constant([0, 1, 0]), num_segments=2) # ==> [[ 4, 6, 6, 4], # [5, 6, 7, 8]]
Si no hay una entrada para un ID de segmento dado i
, genera 1.
Si el ID de segmento dado i
es negativo, entonces el valor correspondiente se descarta y no se incluirá en el resultado.
Argumentos:
- alcance: un objeto de alcance
- segmento_id: un tensor cuya forma es un prefijo de
data.shape
.
Devoluciones:
-
Output
: tiene la misma forma que los datos, excepto por las primeras dimensiones desegment_ids.rank
, que se reemplazan con una única dimensión que tiene el tamañonum_segments
.
Constructores y Destructores | |
---|---|
UnsortedSegmentProd (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input segment_ids, :: tensorflow::Input num_segments) |
Atributos públicos | |
---|---|
operation | |
output |
Funciones publicas | |
---|---|
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 publicas
UnsortedSegmentProd
UnsortedSegmentProd( 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