QuantizedMatMulWithBias

public final class QuantizedMatMulWithBias

Executa uma multiplicação de matriz quantizada de `a` pela matriz` b` com adição de polarização.

As entradas devem ser matrizes bidimensionais e vetor de polarização 1D. E a dimensão interna de `a` (após ser transposta se` transpose_a` for diferente de zero) deve corresponder à dimensão externa de `b` (após ser transposta se` transposed_b` for diferente de zero). Em seguida, faça a operação de adição de broadcast com valores de polarização no resultado da multiplicação da matriz. O tamanho da polarização deve corresponder à dimensão interna de `b`.

Classes aninhadas

aula QuantizedMatMulWithBias.Options Atributos opcionais para QuantizedMatMulWithBias

Métodos Públicos

static <W, T, U, V> QuantizedMatMulWithBias <W>
criar ( Âmbito âmbito, Operando <T> um, Operando <L> b, Operando <V> viés, Operando <flutuador> mina, Operando <flutuador> maxa, Operando <flutuador> minB, Operando <flutuador> maxB, Classe <W > Toutput, Options ... options)
Método de fábrica para criar uma classe envolvendo uma nova operação QuantizedMatMulWithBias.
static QuantizedMatMulWithBias.Options
inputQuantMode (String inputQuantMode)
Resultado <Float>
maxOut ()
O valor flutuante que o valor de saída quantizado mais alto representa.
Resultado <Float>
minOut ()
O valor flutuante que o menor valor de saída quantizado representa.
Saída <W>
fora ()
static QuantizedMatMulWithBias.Options
transposeA (boolean transposeA)
static QuantizedMatMulWithBias.Options
transposeB (boolean transposeB)

Métodos herdados

Métodos Públicos

public static QuantizedMatMulWithBias <W> criar ( Âmbito âmbito, Operando <T> um, Operando <L> b, Operando <V> viés, Operando <flutuador> mina, Operando <flutuador> maxa, Operando <flutuador> minB, Operando <flutuador > maxB, Classe <W> Toutput, Opções ... opções)

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

Parâmetros
alcance escopo atual
uma Uma matriz a ser multiplicada. Deve ser um tensor bidimensional do tipo `quint8`.
b Uma matriz a ser multiplicada e deve ser um tensor bidimensional do tipo `qint8`.
tendência Um tensor de polarização 1D com tamanho correspondente à dimensão interna de `b` (após ser transposto se` transposed_b` for diferente de zero).
minA O valor flutuante que o menor valor `a` quantizado representa.
maxA O valor flutuante que o valor `a` quantizado mais alto representa.
minB O valor flutuante que o menor valor quantizado de `b` representa.
maxB O valor flutuante que o valor `b` quantizado mais alto representa.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de QuantizedMatMulWithBias

public static QuantizedMatMulWithBias.Options inputQuantMode (String inputQuantMode)

Parâmetros
inputQuantMode Modo de quantização de dados de entrada. MIN_FIRST (padrão) ou SCALED.

public Output <Float> maxOut ()

O valor flutuante que o valor de saída quantizado mais alto representa.

public Output <Float> minOut ()

O valor flutuante que o menor valor de saída quantizado representa.

Saída pública <W> out ()

public static QuantizedMatMulWithBias.Options transposeA (Boolean transposeA)

Parâmetros
transporA Se verdadeiro, `a` é transposto antes da multiplicação.

public static QuantizedMatMulWithBias.Options transposeB (Boolean transposeB)

Parâmetros
transporB Se verdadeiro, `b` é transposto antes da multiplicação.