Aplica softmax a um ND `SparseTensor` em lote.
As entradas representam um ND SparseTensor com forma lógica `[..., B, C]` (onde `N> = 2`), e com índices classificados na ordem lexicográfica canônica.
Esta op é equivalente à aplicação normal tf.nn.softmax()
para cada submatriz lógico mais interna com forma `[B, C]`, mas com o trinco que os elementos de zero implicitamente não participar. Especificamente, o algoritmo é equivalente ao seguinte:
(1) Aplica-se tf.nn.softmax()
a uma vista densificada de cada submatriz mais interna com forma `[B, C]`, ao longo da dimensão tamanho-C; (2) mascara as localizações implicitamente zero originais; (3) Renormaliza os elementos restantes.
Portanto, o resultado `SparseTensor` tem exatamente os mesmos índices e forma diferentes de zero.
Constantes
Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
Métodos herdados
boleano | equals (arg0 Object) |
aula final <?> | getClass () |
int | hashCode () |
vazio final | notificar () |
vazio final | notifyAll () |
Fragmento | toString () |
vazio final | wait (longa arg0, int arg1) |
vazio final | wait (arg0 de comprimento) |
vazio final | wait () |
abstrato ExecutionEnvironment | env () Retorne o ambiente de execução em que este op foi criado. |
abstrato Operação |
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Métodos Públicos
pública Output <T> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para as operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
public static SparseSoftmax <T> criar ( Scope escopo, Operando < TInt64 > spIndices, Operando <T> spValues, Operando < TInt64 > spShape)
Método de fábrica para criar uma classe envolvendo uma nova operação SparseSoftmax.
Parâmetros
alcance | escopo atual |
---|---|
spIndices | 2-D. Matriz `NNZ x R` com os índices de valores não vazios em um SparseTensor, em ordem canônica. |
spValues | 1-D. Valores não vazios `NNZ` correspondentes a` sp_indices`. |
spShape | 1-D. Forma da entrada SparseTensor. |
Devoluções
- uma nova instância de SparseSoftmax