Skip to Content
DocumentationNnModulesmlp

mlp

View the code on GitHub 

Structs

Struct: MLP

Fields

  • weights: List[Array]

  • biases: List[Array]

  • hidden_dims: List[Int]

  • num_layers: Int

  • compute_backward: Bool

Methods

__init__(out self, hidden_dims: List[Int], compute_backward: Bool = False)
more details
Args

  • self: Self

  • hidden_dims: List[Int]

  • compute_backward: Bool (default: False)

forward(self, x: Array) -> Array
more details
Args

  • self: Self

  • x: Array

Returns
  • Array
params(self) -> List[Array]
more details
Args
  • self: Self
Returns
  • List[Array]
__str__(self) -> String
more details
Args
  • self: Self
Returns
  • String

Functions

mlp

mlp(args: List[Array]) -> Array
more details
Args
  • args: List[Array]
Returns
  • Array
Last updated on
view_opadagrad