tensorflow:: ops:: ExpandDims

Inserta una dimensión de 1 en la forma de un tensor.

Resumen

Dado un tensor input , esta operación inserta una dimensión de 1 en el axis de índice de dimensión de la forma de input . El axis de índice de dimensión comienza en cero; si se especifica un número negativo para axis , se cuenta hacia atrás desde el final.

Esta operación es útil si desea agregar una dimensión de lote a un solo elemento. Por ejemplo, si tiene una sola imagen con forma [height, width, channels] , puede convertirla en un lote de 1 imagen con expand_dims(image, 0) , lo que hará que tenga la forma [1, height, width, channels] .

Otros ejemplos:

# '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 operación requiere que:

-1-input.dims() <= dim <= input.dims()

Esta operación está relacionada con squeeze() , que elimina dimensiones de tamaño 1.

Argumentos:

• ámbito: Un objeto de ámbito
• eje: 0-D (escalar). Especifica el índice de dimensión en el que se expandirá la forma de input . Debe estar en el rango [-rank(input) - 1, rank(input)] .

Devoluciones:

• Output : Contiene los mismos datos que input , pero su forma tiene una dimensión adicional de tamaño 1 añadida.

Atributos públicos

Operation operation

::tensorflow::Output output

 ExpandDims(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input axis
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const