SparseSegmentMeanGradV2

classe final pública SparseSegmentMeanGradV2

Calcula gradientes para SparseSegmentMean.

Retorna o tensor "saída" com o mesmo formato de grad, exceto para a dimensão 0 cujo valor é o número de índices únicos em "índices". Também retorna o vetor "sorted_unique_indices" contendo os índices correspondentes de "índices".

Métodos Públicos

static <T estende o número, U estende o número, V estende o número> SparseSegmentMeanGradV2 <T, U>
create ( Escopo do escopo , Operando <T> grad, Índices de Operando <U>, Operando <V> segmentIds, Operando <Integer> densoOutputDim0)
Método de fábrica para criar uma classe que envolve uma nova operação SparseSegmentMeanGradV2.
Saída <T>
saída ()
Saída <U>
classificadosUniqueIndices ()

Métodos herdados

Métodos Públicos

public static SparseSegmentMeanGradV2 <T, U> create ( Escopo do escopo , Operando <T> grad, Operando <U> índices, Operando <V> segmentIds, Operando <Integer> densoOutputDim0)

Método de fábrica para criar uma classe que envolve uma nova operação SparseSegmentMeanGradV2.

Parâmetros
escopo escopo atual
graduado gradiente propagado para a operação SparseSegmentMean.
índices índices passados ​​para a operação SparseSegmentMean correspondente.
IDs de segmento segment_ids passado para a operação SparseSegmentMean correspondente.
densoOutputDim0 dimensão 0 de "dados" passados ​​para a operação SparseSegmentMean.
Devoluções
  • uma nova instância de SparseSegmentMeanGradV2

Saída pública <T> saída ()

Saída pública <U> sortedUniqueIndices ()