Hva er nn lineær i PyTorch?
Hva er nn lineær i PyTorch?

Video: Hva er nn lineær i PyTorch?

Video: Hva er nn lineær i PyTorch?
Video: Torch.nn.Linear Module explained 2024, November
Anonim

Fra dokumentasjon: KLASSE lommelykt. nn . Lineær (in_features, out_features, bias=True) Gjelder en lineær transformasjon til innkommende data: y = xW^T + b. Parametere: in_features – størrelsen på hver inndataprøve.

På samme måte spørs det, hvordan fungerer NN lineært?

Lineær . Gjelder en lineær transformasjon til de innkommende dataene, dvs. //y= Ax+b//. Inngangstensoren gitt i forward(input) må enten være en vektor (1D tensor) eller matrise (2D tensor). Hvis inngangen er en matrise, antas hver rad å være en inngangsprøve for gitt batch.

På samme måte, hva er conv2d i PyTorch? konv2d (inngang, vekt, selv. skjevhet, selv. skritt, selv. polstring, selv. utvidelse, selv.

For det andre, hva er PyTorch nn?

PyTorch : nn De nn pakken definerer et sett med moduler, som du kan tenke på som et nevralt nettverkslag som produserer utdata fra input og kan ha noen trenbare vekter. import lommelykt # N er batchstørrelse; D_in er inndatadimensjon; # H er skjult dimensjon; D_out er utgangsdimensjon.

Hvordan bruker du ReLU i PyTorch?

I PyTorch , kan du konstruere en ReLU lag ved hjelp av den enkle funksjonen relu1 = nn. ReLU med argumentet inplace=False. Siden ReLU funksjonen brukes elementmessig, det er ikke nødvendig å spesifisere inngangs- eller utdatadimensjoner. Argumentet på plass bestemmer hvordan funksjonen behandler input.

Anbefalt: