Pytorch - Mean

Vi vil se hvordan du kan returnere gjennomsnittsverdiene fra en tensor ved hjelp av middel () i denne Pytorch -opplæringen.

Pytorch er et open source-ramme tilgjengelig med et Python-programmeringsspråk. Tensor er et flerdimensjonalt utvalg som brukes til å lagre dataene. For å bruke en tensor, må vi importere fakkelmodulen. For å lage en tensor er metoden som brukes tensor ().

Syntaks:

lommelykt.Tensor (data)

Der dataene er en flerdimensjonal matrise.

Mener()

Gjennomsnitt () i Pytorch brukes til å returnere gjennomsnittsverdien av elementer som er til stede i inngangs -tensorobjektet.

Syntaks:

lommelykt.Gjennomsnitt (tensor, dim)

Hvor:

1. Tensoren er inngangstensoren.

2. Dim er å redusere dimensjonen. DIM = 0 spesifiserer kolonnesammenligningen som får gjennomsnittet langs en kolonne og DIM = 1 spesifiserer radens sammenligning som får gjennomsnittet langs raden.

Eksempel 1:

I dette eksemplet vil vi lage en tensor med to dimensjoner som har 3 rader og 5 kolonner og bruke middelfunksjonen () på rader og kolonner.

#import fakkelmodul
Importer fakkel
#Create en tensor med 2 dimensjoner (3 * 5)
#med tilfeldige elementer ved bruk av randn () -funksjon
data = fakkel.Randn (3,5)
#vise
Print (data)
skrive ut()
#get gjennomsnittet langs kolonner med gjennomsnitt ()
trykk ("betyr på tvers av kolonner:")
trykk (fakkel.Gjennomsnitt (data, dim = 0))
skrive ut()
#get gjennomsnittet langs rader med gjennomsnitt ()
trykk ("betyr på tvers av rader:")
trykk (fakkel.Gjennomsnitt (data, dim = 1))

Produksjon:

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]])
Betyr på tvers av kolonner:
Tensor ([-0.3602, 0.4291, 0.2326, 0.2298, -0.5886])
Betyr på tvers av rader:
tensor ([0.4126, -0.0044, -0.4426])

Vi kan se at middelverdiene returneres over kolonnene og radene.

Eksempel 2:

Lag en tensor med 5 * 5 -matrise og returner gjennomsnittet over radene og kolonnene.

#import fakkelmodul
Importer fakkel
#Create en tensor med 2 dimensjoner (5 * 5)
#med tilfeldige elementer ved bruk av randn () -funksjon
data = fakkel.Randn (5,5)
#vise
Print (data)
skrive ut()
#get gjennomsnittet langs kolonner med gjennomsnitt ()
trykk ("betyr på tvers av kolonner:")
trykk (fakkel.Gjennomsnitt (data, dim = 0))
skrive ut()
#get gjennomsnittet langs rader med gjennomsnitt ()
trykk ("betyr på tvers av rader:")
trykk (fakkel.Gjennomsnitt (data, dim = 1))

Produksjon:

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]])
Betyr på tvers av kolonner:
Tensor ([-0.4464, 0.4546, -0.1845, 0.5436, 0.0189])
Betyr på tvers av rader:
tensor ([0.2148, -0.1486, -0.0878, 0.0048, 0.4030])

Vi kan se at middelverdiene over radene og kolonnene ble returnert.

Uten den svake parameteren

Hvis vi ikke spesifiserer DIM -parameteren, returnerer den hele verdens gjennomsnitt.

Eksempel 1:

Lag en 2D -tensor med 5*5 -matrise og returner gjennomsnittsverdien.

#import fakkelmodul
Importer fakkel
#Create en tensor med 2 dimensjoner (5 * 5)
#med tilfeldige elementer ved bruk av randn () -funksjon
data = fakkel.Randn (5,5)
#vise
Print (data)
skrive ut()
#get gjennomsnittet med gjennomsnitt ()
trykk ("Mean:")
trykk (fakkel.Gjennomsnitt (data))

Produksjon:

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]])
Mener :
Tensor (-0.2308)

Eksempel 2:

Lag en 1D -tensor med 5 verdier og returner gjennomsnittsverdien.

#import fakkelmodul
Importer fakkel
#skap en tensor med 5 numeriske verdier
data = fakkel.Tensor ([10.6,20.7,30.6,40.4,50.0])
#vise
Print (data)
skrive ut()
#get gjennomsnittet med gjennomsnitt ()
trykk ("Mean:")
trykk (fakkel.Gjennomsnitt (data))

Produksjon:

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

Arbeid med CPU

Hvis du vil kjøre en Argmax () -funksjon på CPU, må vi lage en tensor med en CPU () -funksjon. Dette kjøres på en CPU -maskin.

Når vi lager en tensor, kan vi bruke CPU () -funksjonen denne gangen.

Syntaks:

lommelykt.Tensor (data).prosessor()

Eksempel 1:

I dette eksemplet vil vi lage en tensor med 2 dimensjoner som har 3 rader og 5 kolonner med CPU () -funksjonen og bruke middelfunksjonen () på rader og kolonner.

#import fakkelmodul
Importer fakkel
#Create en tensor med 2 dimensjoner (3 * 5)
#med tilfeldige elementer ved bruk av randn () -funksjon
data = fakkel.Randn (3,5).prosessor()
#vise
Print (data)
skrive ut()
#get gjennomsnittet langs kolonner med gjennomsnitt ()
trykk ("betyr på tvers av kolonner:")
trykk (fakkel.Gjennomsnitt (data, dim = 0))
skrive ut()
#get gjennomsnittet langs rader med gjennomsnitt ()
trykk ("betyr på tvers av rader:")
trykk (fakkel.Gjennomsnitt (data, dim = 1))

Produksjon:

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]])
Betyr på tvers av kolonner:
Tensor ([-0.3602, 0.4291, 0.2326, 0.2298, -0.5886])
Betyr på tvers av rader:
tensor ([0.4126, -0.0044, -0.4426])

Vi kan se at middelverdiene returneres over kolonnene og radene.

Eksempel 2:

Lag en tensor med 5 * 5 -matrise med CPU () -funksjonen og returner gjennomsnittet over radene og kolonnene.

#import fakkelmodul
Importer fakkel
#Create en tensor med 2 dimensjoner (5 * 5)
#med tilfeldige elementer ved bruk av randn () -funksjon
data = fakkel.Randn (5,5).prosessor()
#vise
Print (data)
skrive ut()
#get gjennomsnittet langs kolonner med gjennomsnitt ()
trykk ("betyr på tvers av kolonner:")
trykk (fakkel.Gjennomsnitt (data, dim = 0))
skrive ut()
#get gjennomsnittet langs rader med gjennomsnitt ()
trykk ("betyr på tvers av rader:")
trykk (fakkel.Gjennomsnitt (data, dim = 1))

Produksjon:

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]])
Betyr på tvers av kolonner:
Tensor ([-0.4464, 0.4546, -0.1845, 0.5436, 0.0189])
Betyr på tvers av rader:
tensor ([0.2148, -0.1486, -0.0878, 0.0048, 0.4030])

Vi kan se at middelverdiene over radene og kolonnene ble returnert.

Konklusjon

I denne Pytorch -leksjonen lærte vi om gjennomsnittlig () -funksjonen og hvordan vi kan bruke den på en tensor for å returnere gjennomsnittsverdiene over kolonnene og radene.

Vi opprettet også en tensor med CPU () -funksjonen og returnerte gjennomsnittsverdiene. Hvis DIM ikke er spesifisert i to eller flerdimensjonal tensor, returnerer den hele verdiene i gjennomsnitt.