Calcula gradientes para SparseSegmentSqrtN.
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> SparseSegmentSqrtNGradV2 <T, U> | |
Saída <T> | saída () |
Saída <U> |
Métodos herdados
Métodos Públicos
public static SparseSegmentSqrtNGradV2 <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 SparseSegmentSqrtNGradV2.
Parâmetros
escopo | escopo atual |
---|---|
graduado | gradiente propagado para a operação SparseSegmentSqrtN. |
índices | índices passados para a operação SparseSegmentSqrtN correspondente. |
IDs de segmento | segment_ids passado para a operação SparseSegmentSqrtN correspondente. |
densoOutputDim0 | dimensão 0 de "dados" passados para SparseSegmentSqrtN op. |
Devoluções
- uma nova instância de SparseSegmentSqrtNGradV2