SparseSegmentSqrtNGradV2

clase final pública SparseSegmentSqrtNGradV2

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>
crear ( alcance alcance, Operando <T> grad, Operando <U> índices, Operando <V> segmentIds, Operando <Integer> densoOutputDim0)
Método de fábrica para crear una clase que envuelve una nueva operación SparseSegmentSqrtNGradV2.
Salida <T>
salida ()
Salida <U>
ordenadosÍndicesUnicos ()

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

Salida pública <T> salida ()

Salida pública <U> sortedUniqueIndices ()