Calcula el producto a lo largo de segmentos 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.
Calcula un tensor tal que \\(output_i = \prod_j data_j\\) donde el producto está sobre `j` tal que `segment_ids[j] == i`.
Si el producto está vacío para un ID de segmento determinado `i`, `salida[i] = 1`.
Nota: Esta operación actualmente solo es compatible con jit_compile=True.
La única diferencia con SegmentProd es la entrada adicional `num_segments`. Esto ayuda a evaluar la forma de salida en tiempo de compilación. `num_segments` debe ser coherente con segment_ids. por ejemplo, Max(segment_ids) - 1 debe ser igual a `num_segments` para un segment_ids 1-d. Con num_segments inconsistentes, la operación aún se ejecuta. La única diferencia es que la salida toma el tamaño de num_segments independientemente del tamaño de segment_ids y datos. para num_segments menores que el tamaño de salida esperado, los últimos elementos se ignoran para num_segments mayores que el tamaño de salida esperado, a los últimos elementos se les asigna 1.
Por ejemplo:
>>> @tf.function(jit_compile=True) ... def prueba(c): ... return tf.raw_ops.SegmentProdV2(data=c, segment_ids=tf.constant([0, 0, 1]), num_segmentos=2) >>> c = tf.constant([[1,2,3,4], [4, 3, 2, 1], [5,6,7,8]]) >>> prueba( c).numpy() matriz([[4, 6, 6, 4], [5, 6, 7, 8]], dtype=int32)
Métodos públicos
| Salida <T> | como salida () Devuelve el identificador simbólico de un tensor. |
| estático <T, U extiende el número, V extiende el número> SegmentProdV2 <T> | crear (alcance alcance , datos del operando <T>, identificadores de segmento del operando <U>, números de segmento del operando <V>) Método de fábrica para crear una clase que envuelve una nueva operación SegmentProdV2. |
| Salida <T> | producción () Tiene la misma forma que los datos, excepto por las primeras dimensiones `segment_ids.rank`, que se reemplazan con una única dimensión que tiene el tamaño `num_segments`. |
Métodos heredados
Métodos públicos
Salida pública <T> como Salida ()
Devuelve el identificador simbólico de un 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.
creación pública estática de SegmentProdV2 <T> (alcance del alcance , datos del operando <T>, identificadores de segmento del operando <U>, números de segmento del operando <V>)
Método de fábrica para crear una clase que envuelve una nueva operación SegmentProdV2.
Parámetros
| alcance | alcance actual |
|---|---|
| identificadores de segmento | Un tensor 1-D cuyo tamaño es igual al tamaño de la primera dimensión de los "datos". Los valores deben ordenarse y pueden repetirse. Los valores deben ser menores que `num_segments`. Precaución: Los valores siempre se validan para ordenarse en la CPU, nunca en la GPU. |
Devoluciones
- una nueva instancia de SegmentProdV2
Salida pública <T> salida ()
Tiene la misma forma que los datos, excepto por las primeras dimensiones `segment_ids.rank`, que se reemplazan con una única dimensión que tiene el tamaño `num_segments`.