Insere uma dimensão de 1 na forma de um tensor.
Dado um tensor `entrada`, esta operação insere uma dimensão de 1 no índice de dimensão` eixo` da forma de `entrada`. O índice de dimensão `eixo` começa em zero; se você especificar um número negativo para o `eixo`, ele será contado para trás 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ê tem uma única imagem com a forma `[altura, largura, canais]`, pode torná-la um lote de 1 imagem com `expand_dims (imagem, 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
Output <T> | asOutput () Retorna o identificador simbólico de um tensor. |
estática <T, U estende Number> ExpandDims <T> | |
Output <T> | saída () Contém os mesmos dados que `entrada`, mas sua forma tem uma dimensão adicional de tamanho 1 adicionada. |
Métodos herdados
Métodos Públicos
pública Output <T> asOutput ()
Retorna o identificador simbólico de um tensor.
As entradas para as 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 ( Scope escopo, Operando <T> entrada, Operando <U> eixo)
Método de fábrica para criar uma classe envolvendo uma nova operação ExpandDims.
Parâmetros
alcance | escopo atual |
---|---|
eixo | 0-D (escalar). Especifica o índice de dimensão no qual expandir a forma de `entrada`. Deve estar no intervalo `[-rank (input) - 1, rank (input)]`. |
Devoluções
- uma nova instância de ExpandDims
pública Output <T> de saída ()
Contém os mesmos dados que `entrada`, mas sua forma tem uma dimensão adicional de tamanho 1 adicionada.