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) - deslocamento` e `x = max(d, 0) - deslocamento`.`offset` é zero, exceto quando o alinhamento da diagonal é para a direita.
offset = max_diag_len - diag_len(d) ; if (`align` in {RIGHT_LEFT, RIGHT_RIGHT}
and `d >= 0`) or
(`align` in {LEFT_RIGHT, RIGHT_RIGHT}
and `d <= 0`)
0 ; otherwise
onde `diag_len(d) = min(cols - max(d, 0), rows + min(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 band from each batch.
tf.matrix_diag_part(input, k = (-1, 2))
==> [[[0, 3, 8], # Output shape: (2, 4, 3)
[2, 7, 6],
[1, 6, 7],
[5, 8, 0]],
[[0, 3, 4],
[4, 3, 8],
[5, 2, 7],
[1, 6, 0]]]
# LEFT_RIGHT alignment.
tf.matrix_diag_part(input, k = (-1, 2), align="LEFT_RIGHT")
==> [[[3, 8, 0], # Output shape: (2, 4, 3)
[2, 7, 6],
[1, 6, 7],
[0, 5, 8]],
[[3, 4, 0],
[4, 3, 8],
[5, 2, 7],
[0, 1, 6]]]
# max_diag_len can be shorter than the main diagonal.
tf.matrix_diag_part(input, k = (-2, -1))
==> [[[5, 8],
[9, 0]],
[[1, 6],
[5, 0]]]
# padding_value = 9
tf.matrix_diag_part(input, k = (1, 3), padding_value = 9)
==> [[[9, 9, 4], # Output shape: (2, 3, 3)
[9, 3, 8],
[2, 7, 6]],
[[9, 9, 2],
[9, 3, 4],
[4, 3, 8]]]
Classes aninhadas
aula | MatrixDiagPartV3.Options | Atributos opcionais para MatrixDiagPartV3 |
Métodos públicos
estático MatrixDiagPartV3.Options | alinhar (alinhar string) |
Saída <T> | comoSaída () Retorna o identificador simbólico de um tensor. |
estático <T> MatrixDiagPartV3 <T> | |
Saída <T> | diagonal () A(s) diagonal(is) extraída(s). |
Métodos Herdados
Métodos públicos
public static MatrixDiagPartV3.Options align (String align)
Parâmetros
alinhar | Algumas diagonais são menores que `max_diag_len` e precisam ser preenchidas. `align` é uma string que especifica como as superdiagonais e as subdiagonais devem ser alinhadas, respectivamente. Há quatro alinhamentos possíveis: "RIGHT_LEFT" (padrão), "LEFT_RIGHT", "LEFT_LEFT" e "RIGHT_RIGHT". "RIGHT_LEFT" alinha superdiagonais à direita (preenche a linha à esquerda) e subdiagonais à esquerda (preenche a linha à direita). É o formato de embalagem que o LAPACK utiliza. cuSPARSE usa "LEFT_RIGHT", que é o alinhamento oposto. |
---|
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 MatrixDiagPartV3 <T> create ( Escopo do escopo, Operando <T> entrada, Operando <Integer> k, Operando <T> paddingValue, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação MatrixDiagPartV3.
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. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de MatrixDiagPartV3