Copie un tensor que ponga a cero todo lo que está fuera de una banda central en cada matriz más interna.
La parte de la `banda` se calcula de la siguiente manera: Suponga 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 del indicador
`en_banda (m, n) = (núm_bajo <0 || (mn) <= núm_bajo)) && (núm_superior <0 || (nm) <= núm_superior)`.
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]]
casos especiales útiles: 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
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
Salida <T> | asOutput () Devuelve el identificador simbólico del tensor. |
Salida <T> | banda () Clasifique el tensor `k` de la misma forma que la entrada. |
estática <T se extiende Ttype , U se extiende TNumber > BandPart <T> |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Métodos públicos
pública de salida <T> asOutput ()
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.
pública de salida <T> banda ()
Clasifique el tensor `k` de la misma forma que la entrada. El tensor en bandas extraído.
public static BandPart <T> crear ( Alcance alcance, operando <T> de entrada, operando <U> numLower, operando <U> numUpper)
Método de fábrica para crear una clase que envuelva una nueva operación BandPart.
Parámetros
alcance | alcance actual |
---|---|
aporte | Tensor de rango `k`. |
numLower | Tensor 0-D. Número de subdiagonales a mantener. Si es negativo, mantenga todo el triángulo inferior. |
numUpper | Tensor 0-D. Número de superdiagonales a mantener. Si es negativo, mantenga el triángulo superior completo. |
Devoluciones
- una nueva instancia de BandPart