Inserta una dimensión de 1 en la forma de un tensor.
Dado un tensor de "entrada", esta operación inserta una dimensión de 1 en el índice de dimensión "eje" de la forma de "entrada". El índice de dimensión `eje` comienza en cero; si especifica un número negativo para el "eje", 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 "[altura, ancho, canales]", puede convertirla en un lote de 1 imagen con "expand_dims (imagen, 0)", lo que hará que la forma "[1, altura , ancho, canales] `.
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.
Métodos públicos
Salida <T> | asOutput () Devuelve el identificador simbólico de un tensor. |
estática <T, U se extiende Número> ExpandDims <T> | |
Salida <T> | salida () Contiene los mismos datos que "input", pero su forma tiene una dimensión adicional de tamaño 1 agregada. |
Métodos heredados
Métodos públicos
pública de salida <T> asOutput ()
Devuelve el identificador simbólico de un tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
public static ExpandDims <T> crear ( Alcance alcance, operando <T> de entrada, operando <U> eje)
Método de fábrica para crear una clase que envuelva una nueva operación ExpandDims.
Parámetros
alcance | alcance actual |
---|---|
eje | 0-D (escalar). Especifica el índice de dimensión en el que expandir la forma de "entrada". Debe estar en el rango `[-rango (entrada) - 1, rango (entrada)]`. |
Devoluciones
- una nueva instancia de ExpandDims
pública de salida <T> salida ()
Contiene los mismos datos que "input", pero su forma tiene una dimensión adicional de tamaño 1 agregada.