Este artículo ejemplificará el método para agregar dimensiones a un tensor en PyTorch.
¿Cómo agregar una dimensión al tensor específico en PyTorch?
Los usuarios pueden agregar dimensiones a cualquier tensor, como un tensor 1D o un tensor 2D en PyTorch. Para agregar nuevas dimensiones a los tensores en una posición específica, consulte los siguientes ejemplos para una mejor comprensión:
- Ejemplo 1: agregar dimensión a un tensor 1D en PyTorch
- Ejemplo 2: agregar dimensión a un tensor 2D en PyTorch
Ejemplo 1: agregar dimensión a un tensor 1D en PyTorch
En este ejemplo, crearemos un tensor 1D y le agregaremos una dimensión en una posición particular. Siga los pasos que se enumeran a continuación para una demostración práctica:
Paso 1: Importar biblioteca
Primero, importe la biblioteca de antorchas:
importar antorcha
Paso 2: crear tensor 1D
Luego, crea un tensor unidimensional. Por ejemplo, hemos creado el siguiente tensor y lo hemos almacenado en el archivo ' X ' variable:
X = antorcha. tensor ( [ 5 , 3 , 8 , 2 ] )
Paso 3: ver la forma del tensor
A continuación, muestre la forma tensor recién creada para ver sus dimensiones:
El siguiente resultado indica que nuestro tensor es unidimensional:
Paso 4: agregar dimensión al tensor 1D
Ahora, utilice el ' torch.unsqueeze(entrada, tenue) 'Función para agregar la dimensión al tensor 1D en la posición específica. Por ejemplo, estamos sumando la dimensión al tensor en el índice 0:
Aquí,
- “ nuevos_decenas ” es la variable que incluye la dimensión agregada.
- “ X ”es el tensor de entrada.
- “ tenue=0 'Se utiliza para agregar dimensión en el índice 0.
Paso 5: verificar la salida
Finalmente, asegúrese de que se haya agregado una nueva dimensión al tensor o no:
En el siguiente resultado, se puede observar que la nueva dimensión se ha agregado al tensor 1D en el índice 0:
Además, los usuarios también pueden agregar dimensiones a otras posiciones. Aquí, hemos agregado la dimensión en el primer índice:
Ejemplo 2: agregar dimensión a un tensor 2D en PyTorch
Aquí, crearemos/haremos un tensor 2D y le agregaremos una dimensión en la posición específica. Pruebe los pasos proporcionados para una implementación práctica:
Paso 1: Importar biblioteca de antorchas
Primero, importe la biblioteca de antorchas:
Paso 2: crea un tensor 2D
Luego, crea un tensor bidimensional. Por ejemplo, hemos creado el siguiente tensor y lo hemos almacenado en el archivo ' X ' variable:
Paso 3: ver la forma del tensor
Después de eso, muestre la forma tensor recién creada para ver sus dimensiones:
Según el siguiente resultado, este tensor es bidimensional:
Paso 4: agregar dimensión al tensor 2D
Ahora, agregue la dimensión al tensor 2D en la posición específica usando el “antorcha.unsqueeze(entrada, tenue)” función. Por ejemplo, estamos sumando la dimensión al tensor en el índice 0:
Paso 5: verificar la salida
Por último, verifique si la nueva dimensión se ha agregado al tensor 2D o no:
El siguiente resultado indica que la nueva dimensión se agregó correctamente al tensor 2D en el índice 0:
Nota : Puede acceder a nuestro Google Colab Notebook en este enlace .
Hemos explicado de manera eficiente el método para agregar dimensión a un tensor en PyTorch usando diferentes ejemplos.
Conclusión
Para agregar dimensiones a un tensor en PyTorch, primero importe la biblioteca de PyTorch. Luego, cree un tensor 1D o 2D y vea sus dimensiones. Después de eso, agregue la dimensión a un tensor en la posición específica usando el botón ' torch.unsqueeze(entrada, tenue) ' función. Los usuarios deben pasar el tensor de entrada y la posición del índice deseada como parámetro para esta función. Este artículo ha ejemplificado el método de agregar dimensiones a un tensor en PyTorch.