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> | salida () |
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