tensorflow :: operaciones :: Expandir

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

Resumen

Dada una input tensor, esta operación inserta una dimensión de 1 en el axis del índice de dimensión de la forma de la input . El axis índice de dimensión comienza en cero; si especifica un número negativo para el 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 de forma [height, width, channels] , puede convertirla en un lote de 1 imagen con expand_dims(image, 0) , lo que hará que 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 las dimensiones de tamaño 1.

Argumentos:

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

Devoluciones:

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

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