clase final pública SparseSegmentMeanGradV2
Calcula gradientes para SparseSegmentMean.
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> SparseSegmentMeanGradV2 <T, U> | |
| Salida <T> | producción () |
| Salida <U> |
Métodos heredados
Métodos públicos
public static SparseSegmentMeanGradV2 <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 SparseSegmentMeanGradV2.
Parámetros
| alcance | alcance actual |
|---|---|
| graduado | gradiente propagado a la operación SparseSegmentMean. |
| índices | índices pasados a la correspondiente operación SparseSegmentMean. |
| identificadores de segmento | segment_ids pasado a la operación SparseSegmentMean correspondiente. |
| densoOutputDim0 | dimensión 0 de "datos" pasada a SparseSegmentMean op. |
Devoluciones
- una nueva instancia de SparseSegmentMeanGradV2