Retorna a parte diagonal em lote de um tensor em lote.
Retorna um tensor com as diagonais `k[0]`-th a `k[1]`-th da `input` em lote.
Suponha que `input` tenha dimensões `r` `[I, J, ..., L, M, N]`. Seja `max_diag_len` o comprimento máximo entre todas as diagonais a serem extraídas, `max_diag_len = min(M + min(k[1], 0), N + min(-k[0], 0))` Seja `num_diags` seja o número de diagonais a serem extraídas, `num_diags = k[1] - k[0] + 1`.
Se `num_diags == 1`, o tensor de saída é de classificação `r - 1` com forma `[I, J, ..., L, max_diag_len]` e valores:
diagonal[i, j, ..., l, n]
= input[i, j, ..., l, n+y, n+x] ; if 0 <= n+y < M and 0 <= n+x < N,
padding_value ; otherwise.
onde `y = max(-k [1], 0)`, `x = max(k[1], 0)`.Caso contrário, o tensor de saída tem classificação `r` com dimensões `[I, J, ..., L, num_diags, max_diag_len]` com valores:
diagonal[i, j, ..., l, m, n]
= input[i, j, ..., l, n+y, n+x] ; if 0 <= n+y < M and 0 <= n+x < N,
padding_value ; otherwise.
onde `d = k[1] - m`, `y = max(-d, 0)` e `x = max(d, 0)`.A entrada deve ser pelo menos uma matriz.
Por exemplo:
input = np.array([[[1, 2, 3, 4], # Input shape: (2, 3, 4)
[5, 6, 7, 8],
[9, 8, 7, 6]],
[[5, 4, 3, 2],
[1, 2, 3, 4],
[5, 6, 7, 8]]])
# A main diagonal from each batch.
tf.matrix_diag_part(input) ==> [[1, 6, 7], # Output shape: (2, 3)
[5, 2, 7]]
# A superdiagonal from each batch.
tf.matrix_diag_part(input, k = 1)
==> [[2, 7, 6], # Output shape: (2, 3)
[4, 3, 8]]
# A tridiagonal band from each batch.
tf.matrix_diag_part(input, k = (-1, 1))
==> [[[2, 7, 6], # Output shape: (2, 3, 3)
[1, 6, 7],
[5, 8, 0]],
[[4, 3, 8],
[5, 2, 7],
[1, 6, 0]]]
# Padding value = 9
tf.matrix_diag_part(input, k = (1, 3), padding_value = 9)
==> [[[4, 9, 9], # Output shape: (2, 3, 3)
[3, 8, 9],
[2, 7, 6]],
[[2, 9, 9],
[3, 4, 9],
[4, 3, 8]]]
Métodos públicos
Saída <T> | comoSaída () Retorna o identificador simbólico de um tensor. |
estático <T> MatrixDiagPartV2 <T> | |
Saída <T> | diagonal () A(s) diagonal(is) extraída(s). |
Métodos Herdados
Métodos públicos
Public Output <T> asOutput ()
Retorna o identificador simbólico de um tensor.
As entradas para 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.
public static MatrixDiagPartV2 <T> create ( Escopo do escopo, Operando <T> input, Operando <Integer> k, Operando <T> paddingValue)
Método de fábrica para criar uma classe que envolve uma nova operação MatrixDiagPartV2.
Parâmetros
alcance | escopo atual |
---|---|
entrada | Classifique o tensor `r` onde `r >= 2`. |
k | Deslocamento(s) diagonal(is). Valor positivo significa superdiagonal, 0 refere-se à diagonal principal e valor negativo significa subdiagonal. `k` pode ser um único inteiro (para uma única diagonal) ou um par de inteiros especificando as extremidades inferior e superior de uma banda de matriz. `k[0]` não deve ser maior que `k[1]`. |
valor de preenchimento | O valor para preencher a área fora da faixa diagonal especificada. O padrão é 0. |
Devoluções
- uma nova instância de MatrixDiagPartV2