Copie um tensor definindo tudo fora de uma banda central em cada matriz mais interna para zero.
A parte `banda` é calculada da seguinte forma: Suponha que` entrada` tenha dimensões `k`` [I, J, K, ..., M, N] `, então a saída é um tensor com a mesma forma onde
`banda [i, j, k, ..., m, n] = in_band (m, n) * entrada [i, j, k, ..., m, n]`.
A função do indicador
`in_band (m, n) = (num_lower <0 || (mn) <= num_lower)) && (num_upper <0 || (nm) <= num_upper)`.
Por exemplo:
# if 'input' is [[ 0, 1, 2, 3]
[-1, 0, 1, 2]
[-2, -1, 0, 1]
[-3, -2, -1, 0]],
tf.matrix_band_part(input, 1, -1) ==> [[ 0, 1, 2, 3]
[-1, 0, 1, 2]
[ 0, -1, 0, 1]
[ 0, 0, -1, 0]],
tf.matrix_band_part(input, 2, 1) ==> [[ 0, 1, 0, 0]
[-1, 0, 1, 0]
[-2, -1, 0, 1]
[ 0, -2, -1, 0]]
tf.matrix_band_part(input, 0, -1) ==> Upper triangular part.
tf.matrix_band_part(input, -1, 0) ==> Lower triangular part.
tf.matrix_band_part(input, 0, 0) ==> Diagonal.
Constantes
| Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
| Output <T> | asOutput () Retorna o identificador simbólico do tensor. |
| Output <T> | banda () Tensor de classificação `k` com a mesma forma da entrada. |
| estática <T estende TType , U estende TNumber > BandPart <T> |
Métodos herdados
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.
pública Output <T> band ()
Tensor de classificação `k` com a mesma forma da entrada. O tensor em faixas extraído.
public static BandPart <T> create ( Scope escopo, Operando <T> entrada, Operando <U> numLower, Operando <U> numUpper)
Método de fábrica para criar uma classe envolvendo uma nova operação BandPart.
Parâmetros
| alcance | escopo atual |
|---|---|
| entrada | Tensor de classificação `k`. |
| numLower | Tensor 0-D. Número de subdiagonais a serem mantidos. Se negativo, mantenha o triângulo inferior inteiro. |
| numUpper | Tensor 0-D. Número de superdiagonais a serem mantidos. Se negativo, mantenha o triângulo superior inteiro. |
Devoluções
- uma nova instância do BandPart