Pytorch - średnia

Zobaczymy, jak zwrócić średnie wartości z tensora za pomocą Mean () w tym samouczku Pytorcha.

Pytorch to open source dostępny z językiem programowania Python. TENSOR to wielowymiarowa tablica używana do przechowywania danych. Aby użyć tensor, musimy zaimportować moduł pochodni. Aby utworzyć tensor, zastosowaną metodą jest tensor ().

Składnia:

latarka.tensor (dane)

Gdzie dane są wielowymiarową tablicą.

Mieć na myśli()

Mean () w Pytorch jest używany do zwrócenia średniej wartości elementów obecnych w obiekcie tensor wejściowych.

Składnia:

latarka.średnia (tensor, DIM)

Gdzie:

1. Tensor to tensor wejściowy.

2. Dim ma zmniejszyć wymiar. Dim = 0 Określa porównanie kolumny, które przenosi średnią wzdłuż kolumny, a DIM = 1 Określa porównanie wierszy, które dostaje średnią wzdłuż wiersza.

Przykład 1:

W tym przykładzie utworzymy tensor z 2 wymiarami, który ma 3 wiersze i 5 kolumn i zastosujemy funkcję średniej () na rzędach i kolumnach.

#Import Moduł pochodni
Importuj pochodnię
#Utwórz tensor z 2 wymiarami (3 * 5)
#z losowymi elementami za pomocą funkcji randn ()
Data = pochodnia.Randn (3,5)
#wyświetlacz
Drukuj (dane)
wydrukować()
#Get Średnia wzdłuż kolumn z Mean ()
Drukuj („Mean w kolumnach:”)
Drukuj (pochodnia.średnia (dane, dim = 0))
wydrukować()
#Get Średnia wzdłuż rzędów z Mean ()
Drukuj („Mean w rzędach:”)
Drukuj (pochodnia.średnia (dane, DIM = 1))

Wyjście:

tensor [[[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]])
Średnia w kolumnach:
tensor ([-0.3602, 0.4291, 0.2326, 0.2298, -0.5886])
Oznaczają rzędy:
tensor ([0.4126, -0.0044, -0.4426])

Widzimy, że wartości średnie są zwracane na kolumnach i wierszach.

Przykład 2:

Utwórz tensor z matrycą 5 * 5 i zwróć średnią na rzędach i kolumnach.

#Import Moduł pochodni
Importuj pochodnię
#Utwórz tensor z 2 wymiarami (5 * 5)
#z losowymi elementami za pomocą funkcji randn ()
Data = pochodnia.Randn (5,5)
#wyświetlacz
Drukuj (dane)
wydrukować()
#Get Średnia wzdłuż kolumn z Mean ()
Drukuj („Mean w kolumnach:”)
Drukuj (pochodnia.średnia (dane, dim = 0))
wydrukować()
#Get Średnia wzdłuż rzędów z Mean ()
Drukuj („Mean w rzędach:”)
Drukuj (pochodnia.średnia (dane, DIM = 1))

Wyjście:

tensor ([[-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]])
Średnia w kolumnach:
tensor ([-0.4464, 0.4546, -0.1845, 0.5436, 0.0189])
Oznaczają rzędy:
tensor ([0.2148, -0.1486, -0.0878, 0.0048, 0.4030])

Widzimy, że średnie wartości w rzędach i kolumnach zostały zwrócone.

Bez parametru Dim

Jeśli nie określamy parametru DIM, zwraca średnią całej wartości.

Przykład 1:

Utwórz tensor 2D z matrycą 5*5 i zwróć średnią wartość.

#Import Moduł pochodni
Importuj pochodnię
#Utwórz tensor z 2 wymiarami (5 * 5)
#z losowymi elementami za pomocą funkcji randn ()
Data = pochodnia.Randn (5,5)
#wyświetlacz
Drukuj (dane)
wydrukować()
#Get Średnia ze średnią ()
Drukuj („Mean:”)
Drukuj (pochodnia.średnia (dane))

Wyjście:

tensor ([[-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]])
Mieć na myśli :
tensor (-0.2308)

Przykład 2:

Utwórz tensor 1D z 5 wartościami i zwróć średnią wartość.

#Import Moduł pochodni
Importuj pochodnię
#Utworz tensor z 5 wartościami liczbowymi
Data = pochodnia.tensor [[10.6,20.7,30.6,40.4,50.0])
#wyświetlacz
Drukuj (dane)
wydrukować()
#Get Średnia ze średnią ()
Drukuj („Mean:”)
Drukuj (pochodnia.średnia (dane))

Wyjście:

tensor [[10.6000, 20.7000, 30.6000, 40.4000, 50.0000])
Mieć na myśli :
tensor (30.4600)

Pracuj z CPU

Jeśli chcesz uruchomić funkcję argmax () na procesorze, musimy utworzyć tensor z funkcją cpu (). Będzie to działać na komputerze procesora.

Kiedy tworzymy tensor, tym razem możemy użyć funkcji CPU ().

Składnia:

latarka.tensor (dane).procesor()

Przykład 1:

W tym przykładzie utworzymy tensor z 2 wymiarami, który ma 3 wiersze i 5 kolumn z funkcją CPU () i zastosujemy funkcję średnią () na rzędach i kolumnach.

#Import Moduł pochodni
Importuj pochodnię
#Utwórz tensor z 2 wymiarami (3 * 5)
#z losowymi elementami za pomocą funkcji randn ()
Data = pochodnia.Randn (3,5).procesor()
#wyświetlacz
Drukuj (dane)
wydrukować()
#Get Średnia wzdłuż kolumn z Mean ()
Drukuj („Mean w kolumnach:”)
Drukuj (pochodnia.średnia (dane, dim = 0))
wydrukować()
#Get Średnia wzdłuż rzędów z Mean ()
Drukuj („Mean w rzędach:”)
Drukuj (pochodnia.średnia (dane, DIM = 1))

Wyjście:

tensor [[[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]])
Średnia w kolumnach:
tensor ([-0.3602, 0.4291, 0.2326, 0.2298, -0.5886])
Oznaczają rzędy:
tensor ([0.4126, -0.0044, -0.4426])

Widzimy, że wartości średnie są zwracane na kolumnach i wierszach.

Przykład 2:

Utwórz tensor z matrycą 5 * 5 z funkcją CPU () i zwróć średnią na rzędach i kolumnach.

#Import Moduł pochodni
Importuj pochodnię
#Utwórz tensor z 2 wymiarami (5 * 5)
#z losowymi elementami za pomocą funkcji randn ()
Data = pochodnia.Randn (5,5).procesor()
#wyświetlacz
Drukuj (dane)
wydrukować()
#Get Średnia wzdłuż kolumn z Mean ()
Drukuj („Mean w kolumnach:”)
Drukuj (pochodnia.średnia (dane, dim = 0))
wydrukować()
#Get Średnia wzdłuż rzędów z Mean ()
Drukuj („Mean w rzędach:”)
Drukuj (pochodnia.średnia (dane, DIM = 1))

Wyjście:

tensor ([[-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]])
Średnia w kolumnach:
tensor ([-0.4464, 0.4546, -0.1845, 0.5436, 0.0189])
Oznaczają rzędy:
tensor ([0.2148, -0.1486, -0.0878, 0.0048, 0.4030])

Widzimy, że średnie wartości w rzędach i kolumnach zostały zwrócone.

Wniosek

W tej lekcji Pytorcha dowiedzieliśmy się o funkcji Mean () i jak zastosować ją na tensor, aby zwrócić średnie wartości w kolumnach i wierszach.

Stworzyliśmy również tensor z funkcją CPU () i zwróciliśmy średnie wartości. Jeśli DIM nie jest określony w dwóch lub wielowymiarowych tensor, zwraca średnio całe wartości.