Insere uma dimensão de 1 na forma de um tensor.
Dado um tensor `input`, esta operação insere uma dimensão 1 no índice de dimensão `axis` da forma `input`. O índice de dimensão `axis` começa em zero; se você especificar um número negativo para `eixo`, ele será contado regressivamente a partir do final.
Esta operação é útil se você deseja adicionar uma dimensão de lote a um único elemento. Por exemplo, se você tiver uma única imagem da forma `[height, width, pipes]`, poderá torná-la um lote de 1 imagem com `expand_dims(image, 0)`, o que tornará a forma `[1, height , largura, canais]`.
Outros exemplos:
# 't' is a tensor of shape [2]
shape(expand_dims(t, 0)) ==> [1, 2]
shape(expand_dims(t, 1)) ==> [2, 1]
shape(expand_dims(t, -1)) ==> [2, 1]
# 't2' is a tensor of shape [2, 3, 5]
shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]
Esta operação requer que:`-1-input.dims() <= dim <= input.dims()`
Esta operação está relacionada a `squeeze()`, que remove dimensões de tamanho 1.
Constantes
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
Saída <T> | asOutput () Retorna o identificador simbólico do tensor. |
estático <T estende TType > ExpandDims <T> | criar (escopo do escopo , entrada do operando <T>, operando <? estende o eixo TNumber >) Método de fábrica para criar uma classe que envolve uma nova operação ExpandDims. |
Saída <T> | saída () Contém os mesmos dados que `input`, mas sua forma tem uma dimensão adicional de tamanho 1 adicionada. |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
Saída pública <T> asOutput ()
Retorna o identificador simbólico do 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 ExpandDims <T> create ( escopo do escopo , entrada do operando <T>, operando <? estende TNumber > eixo)
Método de fábrica para criar uma classe que envolve uma nova operação ExpandDims.
Parâmetros
escopo | escopo atual |
---|---|
eixo | 0-D (escalar). Especifica o índice de dimensão no qual expandir a forma de `input`. Deve estar no intervalo `[-rank(input) - 1, rank(input)]`. |
Devoluções
- uma nova instância de ExpandDims
Saída pública <T> saída ()
Contém os mesmos dados que `input`, mas sua forma tem uma dimensão adicional de tamanho 1 adicionada.