PyTorch es una biblioteca de aprendizaje automático que permite a los usuarios trabajar con tensores. Los tensores son estructuras de datos esenciales que pueden tener diferentes tipos de datos y formas. A veces, es posible que los usuarios quieran realizar la operación de expansión en tensores para ampliar su tamaño. La operación de expansión permite a los usuarios repetir un tensor a lo largo de dimensiones específicas. PyTorch proporciona el atributo 'expand()' que toma un tensor y una lista de tamaños como entradas. Devuelve un nuevo tensor que tiene los mismos datos pero con diferentes dimensiones.
Este artículo ilustrará el método para utilizar la operación de expansión en tensores en PyTorch.
¿Cómo utilizar la operación de expansión en PyTorch?
Para utilizar la operación de expansión en PyTorch, consulte los pasos que se proporcionan a continuación:
- Importar biblioteca PyTorch
- Crea un tensor deseado
- Ver el tamaño del tensor de entrada
- Expande el tensor usando el botón ' expandir() ' atributo
- Mostrar tensor expandido y su tamaño.
Paso 1: importar la biblioteca PyTorch
Primero, importe el ' antorcha ”biblioteca para usar la operación de expansión:
importar antorcha
Paso 2: crea un tensor
Luego, cree el tensor deseado usando el botón ' antorcha.tensor() ”funciona e imprime sus elementos. Aquí, estamos creando lo siguiente ' decenas 'tensor:
decenas = antorcha. tensor ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
imprimir ( decenas )
En el siguiente resultado, el tensor se creó correctamente:
Paso 3: Ver el tamaño del tensor de entrada
A continuación, vea el tamaño del “creado anteriormente” decenas ” tensor usando el “ tamaño() ' atributo:
imprimir ( 'Tamaño del tensor:' , Decenas. tamaño ( ) )Según el siguiente resultado, el tamaño del tensor es 3×1:
Paso 4: expandir el tensor
Ahora, usa el ' expandir() ”atributo para realizar la operación de expansión y extender el tensor a una nueva dimensión. Es necesario proporcionar el tensor y la lista de tamaños como entrada. Aquí, estamos expandiendo el tensor a un tamaño de 3×4:
Exp_decenas = Decenas. expandir ( 3 , 4 )Paso 5: mostrar el tensor expandido y su tamaño
Por último, imprima los elementos del tensor expandido y su tamaño:
imprimir ( Exp_decenas )imprimir ( Exp_tens. tamaño ( ) )
El siguiente resultado muestra el tensor expandido y su tamaño, es decir, 3 × 4. Esto indica que la operación ampliada se ha realizado correctamente:
Hemos explicado el método eficiente para utilizar la operación de expansión en PyTorch.
Nota : Puede acceder a nuestro Google Colab Notebook en este enlace .
Conclusión
Para utilizar la operación de expansión en PyTorch, primero importe la biblioteca de la antorcha. Luego, cree el tensor deseado y vea sus elementos y tamaño. A continuación, utilice el botón ' expandir() 'Atributo para expandir el tensor de entrada. Por último, imprima el tensor expandido y vea su tamaño. Este artículo ha demostrado el método para utilizar la operación de expansión en tensores en PyTorch.