Copie un tensor poniendo a cero todo lo que está fuera de una banda central en cada matriz más interna.
La parte `banda` se calcula de la siguiente manera: Supongamos que la `entrada` tiene `k` dimensiones `[I, J, K, ..., M, N]`, entonces la salida es un tensor con la misma forma donde
`banda [i, j, k, ..., m, n] = in_band (m, n) * entrada [i, j, k, ..., m, n]`.
La función indicadora
`in_band(m, n) = (num_upper < 0 || (mn) <= num_lower)) && (num_upper < 0 || (nm) <= num_upper)`.
Por ejemplo:
# 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
| Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
| Salida <T> | como salida () Devuelve el identificador simbólico del tensor. |
| Salida <T> | banda () Tensor de rango `k` de la misma forma que la entrada. |
| estático <T extiende TType , U extiende TNumber > BandPart <T> |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
Salida pública <T> como Salida ()
Devuelve el identificador simbólico del tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
Banda de salida pública <T> ()
Tensor de rango `k` de la misma forma que la entrada. El tensor bandeado extraído.
creación pública estática de BandPart <T> (alcance del alcance, entrada del operando <T>, operando <U> numLower, operando <U> numUpper)
Método de fábrica para crear una clase que envuelve una nueva operación BandPart.
Parámetros
| alcance | alcance actual |
|---|---|
| aporte | Tensor de rango `k`. |
| numInferior | Tensor 0-D. Número de subdiagonales a mantener. Si es negativo, conserve todo el triángulo inferior. |
| numSuperior | Tensor 0-D. Número de superdiagonales a mantener. Si es negativo, conserve todo el triángulo superior. |
Devoluciones
- una nueva instancia de BandPart