Calcula gradientes para SparseSegmentSum.
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> SparseSegmentSumGradV2 <T, U> | |
Salida <T> | producción () |
Salida <U> |
Métodos heredados
Métodos públicos
public static SparseSegmentSumGradV2 <T, U> create (alcance del alcance , operando <T> grad, índices del operando <U>, operando <V> segmentIds, operando <Integer> densoOutputDim0)
Método de fábrica para crear una clase que envuelve una nueva operación SparseSegmentSumGradV2.
Parámetros
alcance | alcance actual |
---|---|
graduado | gradiente propagado a la operación SparseSegmentSum. |
índices | índices pasados a la correspondiente operación SparseSegmentSum. |
identificadores de segmento | segment_ids pasado a la operación SparseSegmentSum correspondiente. |
densoOutputDim0 | dimensión 0 de "datos" pasada a SparseSegmentSum op. |
Devoluciones
- una nueva instancia de SparseSegmentSumGradV2