Copia un tensore impostando a zero tutto ciò che è al di fuori di una banda centrale in ciascuna matrice più interna.
La parte "banda" viene calcolata come segue: Supponiamo che "input" abbia "k" dimensioni "[I, J, K, ..., M, N]", quindi l'output è un tensore con la stessa forma in cui
`band[i, j, k, ..., m, n] = in_band(m, n) * input[i, j, k, ..., m, n]`.
La funzione dell'indicatore
`in_band(m, n) = (num_inferiore < 0 || (mn) <= num_inferiore)) && (num_superiore < 0 || (nm) <= num_superiore)`.
Per esempio:
# 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.
Costanti
Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi pubblici
Metodi ereditati
booleano finale | è uguale a (Oggetto oggetto) |
finale int | codicehash () |
Operazione | |
stringa finale | aStringa () |
booleano | è uguale a (Oggetto arg0) |
Classe finale<?> | getClass () |
int | codicehash () |
vuoto finale | avvisare () |
vuoto finale | notificaTutti () |
Corda | aStringa () |
vuoto finale | attendere (lungo arg0, int arg1) |
vuoto finale | aspetta (lungo arg0) |
vuoto finale | Aspettare () |
ambiente di esecuzione astratto | ambiente () Restituisce l'ambiente di esecuzione in cui è stata creata questa operazione. |
Operazione astratta |
risultato astratto <T> | comeuscita () Restituisce l'handle simbolico del tensore. |
astratto T | come tensore () Restituisce il tensore in questo operando. |
forma astratta | |
Classe astratta<T> | tipo () Restituisce il tipo tensore di questo operando |
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
Metodi pubblici
Uscita pubblica <T> asOutput ()
Restituisce l'handle simbolico del tensore.
Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.
banda di uscita <T> pubblica ()
Tensore di rango `k` della stessa forma dell'input. Il tensore a banda estratto.
public static BandPart <T> create ( ambito ambito , operando <T> input, operando <U> numLower, operando <U> numUpper)
Metodo factory per creare una classe che racchiude una nuova operazione BandPart.
Parametri
portata | ambito attuale |
---|---|
ingresso | Tensore di rango `k`. |
numeroInferiore | Tensore 0-D. Numero di sottodiagonali da conservare. Se negativo, mantieni l'intero triangolo inferiore. |
numUpper | Tensore 0-D. Numero di superdiagonali da conservare. Se negativo, mantieni l'intero triangolo superiore. |
Ritorni
- una nuova istanza di BandPart