SparseSegmentSqrtNGradV2

classe final pública SparseSegmentSqrtNGradV2

Calcula gradientes para SparseSegmentSqrtN.

Retorna o tensor "saída" com o mesmo formato de grad, exceto para a dimensão 0 cujo valor é o número de índices únicos em "índices". Também retorna o vetor "sorted_unique_indices" contendo os índices correspondentes de "índices".

Métodos Públicos

static <T estende o número, U estende o número, V estende o número> SparseSegmentSqrtNGradV2 <T, U>
create ( Escopo do escopo , Operando <T> grad, Índices de Operando <U>, Operando <V> segmentIds, Operando <Integer> densoOutputDim0)
Método de fábrica para criar uma classe que envolve uma nova operação SparseSegmentSqrtNGradV2.
Saída <T>
saída ()
Saída <U>
classificadosUniqueIndices ()

Métodos herdados

Métodos Públicos

public static SparseSegmentSqrtNGradV2 <T, U> create ( Escopo do escopo , Operando <T> grad, Operando <U> índices, Operando <V> segmentIds, Operando <Integer> densoOutputDim0)

Método de fábrica para criar uma classe que envolve uma nova operação SparseSegmentSqrtNGradV2.

Parâmetros
escopo escopo atual
graduado gradiente propagado para a operação SparseSegmentSqrtN.
índices índices passados ​​para a operação SparseSegmentSqrtN correspondente.
IDs de segmento segment_ids passado para a operação SparseSegmentSqrtN correspondente.
densoOutputDim0 dimensão 0 de "dados" passados ​​para SparseSegmentSqrtN op.
Devoluções
  • uma nova instância de SparseSegmentSqrtNGradV2

Saída pública <T> saída ()

Saída pública <U> sortedUniqueIndices ()