Commit 93b3b14

author

HP Z4 G4

committed

init

1 parent 6102cf6 commit 93b3b14Copy full SHA for 93b3b14

File tree

+22

-0

lines changed

+22

-0

lines changed

Lines changed: 22 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,22 @@`
	`1`	`+import torch`
	`2`	`+import torch.nn as nn`
	`3`	`+`
	`4`	`+class LinearModel(nn.Module):`
	`5`	`+ def __init__(self, ndim):`
	`6`	`+ super(LinearModel, self).__init__()`
	`7`	`+ self.ndim = ndim`
	`8`	`+`
	`9`	`+ self.weight = nn.Parameter(torch.randn(ndim, 1))`
	`10`	`+ self.bias = nn.Parameter(torch.randn(1))`
	`11`	`+`
	`12`	`+ def forward(self, x):`
	`13`	`+ return x.mm(self.weight) + self.bias`
	`14`	`+`
	`15`	`+lm = LinearModel(5)`
	`16`	`+x = torch.randn(4, 5)`
	`17`	`+`
	`18`	`+print(x)`
	`19`	`+print(lm(x))`
	`20`	`+`
	`21`	`+print(list(lm.named_parameters()))`
	`22`	`+print(list(lm.parameters()))`

Comments

(0)