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 de segment_ids.rank , que se reemplazan con una única dimensión que tiene el tamaño num_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