tensorflow :: ops :: UnsortedSegmentProd
#include <math_ops.h>
Calcula o produto ao longo de segmentos de um tensor.
Resumo
Leia a seção sobre segmentação para obter uma explicação dos segmentos.
Este operador é semelhante ao operador de soma de segmento não classificado encontrado (aqui) . Em vez de calcular a soma dos segmentos, ele calcula o produto de todas as entradas pertencentes a um segmento de forma que:
\(output_i = {j...} data[j...]\) em que o produto está acima das tuplas j...
modo que segment_ids[j...] == i
.
Por exemplo:
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]]
Se não houver nenhuma entrada para um determinado ID de segmento i
, ele gera 1.
Se o ID de segmento i
fornecido for negativo, o valor correspondente será descartado e não será incluído no resultado.
Argumentos:
- escopo: um objeto Scope
- segment_ids: um tensor cuja forma é um prefixo de
data.shape
.
Retorna:
-
Output
: tem a mesma forma que os dados, exceto para as primeiras dimensõessegment_ids.rank
, que são substituídas por uma única dimensão que tem o tamanhonum_segments
.
Construtores e Destruidores | |
---|---|
UnsortedSegmentProd (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input segment_ids, :: tensorflow::Input num_segments) |
Atributos públicos | |
---|---|
operation | |
output |
Funções públicas | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Atributos públicos
Operação
Operation operation
resultado
::tensorflow::Output output
Funções públicas
UnsortedSegmentProd
UnsortedSegmentProd( const ::tensorflow::Scope & scope, ::tensorflow::Input data, ::tensorflow::Input segment_ids, ::tensorflow::Input num_segments )
nó
::tensorflow::Node * node() const
operador :: tensorflow :: Input
operator::tensorflow::Input() const
operador :: tensorflow :: Saída
operator::tensorflow::Output() const