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> | |
Saída <T> | saída () |
Saída <U> |
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