Calcula gradientes para SparseSegmentSqrtN.
Devuelve la "salida" del tensor con la misma forma que grad, excepto la dimensión 0 cuyo valor es el número de índices únicos en "índices". También devuelve el vector "sorted_unique_indices" que contiene los índices correspondientes de "indices".
Métodos públicos
estático <T extiende el número, U extiende el número, V extiende el número> SparseSegmentSqrtNGradV2 <T, U> | |
Salida <T> | salida () |
Salida <U> |
Métodos heredados
Métodos públicos
público estático SparseSegmentSqrtNGradV2 <T, U> crear (alcance alcance , operando <T> grad, índices operando <U>, operando <V> segmentIds, operando <Integer> densoOutputDim0)
Método de fábrica para crear una clase que envuelve una nueva operación SparseSegmentSqrtNGradV2.
Parámetros
alcance | alcance actual |
---|---|
graduado | gradiente propagado a la operación SparseSegmentSqrtN. |
índices | índices pasados a la correspondiente operación SparseSegmentSqrtN. |
identificadores de segmento | segment_ids pasado a la operación SparseSegmentSqrtN correspondiente. |
densoOutputDim0 | dimensión 0 de "datos" pasada a SparseSegmentSqrtN op. |
Devoluciones
- una nueva instancia de SparseSegmentSqrtNGradV2