tensorflow :: operaciones :: Expandir

#include <array_ops.h>

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.

Constructores y Destructores

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

Atributos públicos

operation
output

Funciones publicas

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Atributos públicos

operación

Operation operation

producción

::tensorflow::Output output

Funciones publicas

Expandir

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

nodo

::tensorflow::Node * node() const

operador :: tensorflow :: Entrada

 operator::tensorflow::Input() const

operador :: tensorflow :: Salida

 operator::tensorflow::Output() const