ExpandDims

classe final pública ExpandDims

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 com a forma `[altura, largura, canais]`, poderá torná-la um lote de 1 imagem com `expand_dims(image, 0)`, o que tornará a forma `[1, altura , 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.

Métodos Públicos

Saída <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
estático <T, U estende número> ExpandDims <T>
criar (escopo do escopo , entrada do operando <T>, eixo do operando <U>)
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

Métodos Públicos

Saída pública <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 ExpandDims <T> create (escopo do escopo , entrada do operando <T>, eixo do operando <U>)

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.