MCPcopy Create free account
hub / github.com/RolnickLab/climart / forward

Method forward

climart/models/modules/mlp.py:63–68  ·  view source on GitHub ↗
(self, X: Tensor)

Source from the content-addressed store, hash-verified

61 self.out_layer = nn.Sequential(*out_layer)
62
63 def forward(self, X: Tensor) -> Tensor:
64 for layer in self.hidden_layers:
65 X = layer(X)
66
67 Y = self.out_layer(X)
68 return Y.squeeze(1)
69
70
71class MLP_Block(nn.Module):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected