Pytorch - moyen

Nous verrons comment renvoyer les valeurs moyennes d'un tenseur en utilisant Mean () dans ce tutoriel pytorch.

Pytorch est un cadre open source disponible avec un langage de programmation Python. Tensor est un tableau multidimensionnel utilisé pour stocker les données. Pour utiliser un tenseur, nous devons importer le module de torche. Pour créer un tenseur, la méthode utilisée est tenseur ().

Syntaxe:

torche.tenseur (données)

Où les données sont un tableau multidimensionnel.

Moyenne()

Mean () dans Pytorch est utilisé pour renvoyer la valeur moyenne des éléments présents dans l'objet tenseur d'entrée.

Syntaxe:

torche.moyenne (tenseur, dim)

Où:

1. Le tenseur est le tenseur d'entrée.

2. DIM est de réduire la dimension. DIM = 0 Spécifie la comparaison de la colonne qui obtient la moyenne le long d'une colonne et DIM = 1 spécifie la comparaison des lignes qui obtient la moyenne le long de la ligne.

Exemple 1:

Dans cet exemple, nous créerons un tenseur avec 2 dimensions qui a 3 lignes et 5 colonnes et appliquer la fonction moyenne () sur les lignes et les colonnes.

Module de torche #mport
Importer une torche
#create un tenseur avec 2 dimensions (3 * 5)
# avec des éléments aléatoires en utilisant la fonction randn ()
données = torche.Randn (3,5)
#afficher
Imprimer (données)
imprimer()
#get moyen le long des colonnes avec moyenne ()
Imprimer ("moyenne entre les colonnes:")
imprimer (torche.moyenne (données, dim = 0))
imprimer()
#get moyen le long des lignes avec moyenne ()
Imprimer ("moyenne entre les lignes:")
imprimer (torche.moyenne (données, dim = 1))

Sortir:

tenseur ([[1.5484, 1.4450, 0.5954, -0.1447, -1.3809],
[-0.9090, -0.6124, 0.4644, 0.3485, 0.6863],
[-1.7201, 0.4546, -0.3618, 0.4858, -1.0712]])
Moyenne à travers les colonnes:
Tensor ([- 0.3602, 0.4291, 0.2326, 0.2298, -0.5886])
Moyenne à travers les lignes:
Tensor ([0.4126, -0.0044, -0.4426])

Nous pouvons voir que les valeurs moyennes sont renvoyées à travers les colonnes et les lignes.

Exemple 2:

Créez un tenseur avec 5 * 5 matrice et renvoyez la moyenne à travers les lignes et les colonnes.

Module de torche #mport
Importer une torche
#create un tenseur avec 2 dimensions (5 * 5)
# avec des éléments aléatoires en utilisant la fonction randn ()
données = torche.Randn (5,5)
#afficher
Imprimer (données)
imprimer()
#get moyen le long des colonnes avec moyenne ()
Imprimer ("moyenne entre les colonnes:")
imprimer (torche.moyenne (données, dim = 0))
imprimer()
#get moyen le long des lignes avec moyenne ()
Imprimer ("moyenne entre les lignes:")
imprimer (torche.moyenne (données, dim = 1))

Sortir:

tenseur ([[- 1.8994, 0.2208, -0.0023, 1.9119, 0.8428],
[-1.4042, -0.9700, 0.4683, 1.5860, -0.4229],
[-0.5011, 1.7210, -0.0949, -0.8114, -0.7528],
[0.1496, 0.4154, -0.5784, 0.2983, -0.2608],
[ 1.4232, 0.8856, -0.7154, -0.2667, 0.6884]])
Moyenne à travers les colonnes:
Tensor ([- 0.4464, 0.4546, -0.1845, 0.5436, 0.0189])
Moyenne à travers les lignes:
Tensor ([0.2148, -0.1486, -0.0878, 0.0048, 0.4030])

Nous pouvons voir que les valeurs moyennes entre les lignes et les colonnes ont été renvoyées.

Sans le paramètre DIM

Si nous ne spécifions pas le paramètre DIM, il renvoie la moyenne de la valeur entière.

Exemple 1:

Créez un tenseur 2D avec 5 * 5 matrice et renvoyez la valeur moyenne.

Module de torche #mport
Importer une torche
#create un tenseur avec 2 dimensions (5 * 5)
# avec des éléments aléatoires en utilisant la fonction randn ()
données = torche.Randn (5,5)
#afficher
Imprimer (données)
imprimer()
#get moyen avec moyenne ()
Imprimer ("Mean:")
imprimer (torche.moyenne (données))

Sortir:

tenseur ([[- 1.3824, 0.5979, 0.0170, -0.1703, -0.9783],
[-0.5721, -1.0704, -0.7148, -1.4605, 0.1514],
[-1.5455, 1.5261, 1.3712, -1.3692, -1.0385],
[ 1.0159, 0.0484, -0.4317, -1.3518, 0.9220],
[-1.5225, 0.5126, -0.2473, 0.8433, 1.0807]])
Moyenne :
tenseur (-0.2308)

Exemple 2:

Créez un tenseur 1D avec 5 valeurs et renvoyez la valeur moyenne.

Module de torche #mport
Importer une torche
#create un tenseur avec 5 valeurs numériques
données = torche.Tensor ([10.6,20.7,30.6,40.4,50.0])
#afficher
Imprimer (données)
imprimer()
#get moyen avec moyenne ()
Imprimer ("Mean:")
imprimer (torche.moyenne (données))

Sortir:

Tensor ([10.6000, 20.7000, 30.6000, 40.4000, 50.0000])
Moyenne :
tenseur (30.4600)

Travailler avec le processeur

Si vous souhaitez exécuter une fonction argmax () sur le CPU, nous devons créer un tenseur avec une fonction CPU (). Cela fonctionnera sur une machine CPU.

Lorsque nous créons un tenseur, cette fois, nous pouvons utiliser la fonction CPU ().

Syntaxe:

torche.tenseur (données).CPU()

Exemple 1:

Dans cet exemple, nous créerons un tenseur avec 2 dimensions qui a 3 lignes et 5 colonnes avec la fonction CPU () et appliquer la fonction moyenne () sur les lignes et les colonnes.

Module de torche #mport
Importer une torche
#create un tenseur avec 2 dimensions (3 * 5)
# avec des éléments aléatoires en utilisant la fonction randn ()
données = torche.Randn (3,5).CPU()
#afficher
Imprimer (données)
imprimer()
#get moyen le long des colonnes avec moyenne ()
Imprimer ("moyenne entre les colonnes:")
imprimer (torche.moyenne (données, dim = 0))
imprimer()
#get moyen le long des lignes avec moyenne ()
Imprimer ("moyenne entre les lignes:")
imprimer (torche.moyenne (données, dim = 1))

Sortir:

tenseur ([[1.5484, 1.4450, 0.5954, -0.1447, -1.3809],
[-0.9090, -0.6124, 0.4644, 0.3485, 0.6863],
[-1.7201, 0.4546, -0.3618, 0.4858, -1.0712]])
Moyenne à travers les colonnes:
Tensor ([- 0.3602, 0.4291, 0.2326, 0.2298, -0.5886])
Moyenne à travers les lignes:
Tensor ([0.4126, -0.0044, -0.4426])

Nous pouvons voir que les valeurs moyennes sont renvoyées à travers les colonnes et les lignes.

Exemple 2:

Créez un tenseur avec 5 * 5 matrice avec la fonction CPU () et renvoyez la moyenne à travers les lignes et les colonnes.

Module de torche #mport
Importer une torche
#create un tenseur avec 2 dimensions (5 * 5)
# avec des éléments aléatoires en utilisant la fonction randn ()
données = torche.Randn (5,5).CPU()
#afficher
Imprimer (données)
imprimer()
#get moyen le long des colonnes avec moyenne ()
Imprimer ("moyenne entre les colonnes:")
imprimer (torche.moyenne (données, dim = 0))
imprimer()
#get moyen le long des lignes avec moyenne ()
Imprimer ("moyenne entre les lignes:")
imprimer (torche.moyenne (données, dim = 1))

Sortir:

tenseur ([[- 1.8994, 0.2208, -0.0023, 1.9119, 0.8428],
[-1.4042, -0.9700, 0.4683, 1.5860, -0.4229],
[-0.5011, 1.7210, -0.0949, -0.8114, -0.7528],
[0.1496, 0.4154, -0.5784, 0.2983, -0.2608],
[ 1.4232, 0.8856, -0.7154, -0.2667, 0.6884]])
Moyenne à travers les colonnes:
Tensor ([- 0.4464, 0.4546, -0.1845, 0.5436, 0.0189])
Moyenne à travers les lignes:
Tensor ([0.2148, -0.1486, -0.0878, 0.0048, 0.4030])

Nous pouvons voir que les valeurs moyennes entre les lignes et les colonnes ont été renvoyées.

Conclusion

Dans cette leçon de pytorch, nous avons appris la fonction moyenne () et comment l'appliquer sur un tenseur pour renvoyer les valeurs moyennes à travers les colonnes et les lignes.

Nous avons également créé un tenseur avec la fonction CPU () et retourné les valeurs moyennes. Si le DIM n'est pas spécifié dans le tenseur à deux ou multidimensionnels, il renvoie des valeurs entières en moyenne.