Collect parameters from modules

Collect parameters from modules

Usage

mlx_parameters(module)

Arguments

module

An mlx_module or list of modules.

Value

A list of mlx_param objects.

See also

mlx.nn.Module.parameters

Examples

set.seed(1)
layer <- mlx_linear(2, 1)
mlx_parameters(layer)
#> [[1]]
#> $env
#> <environment: 0x148637b88>
#> 
#> $name
#> [1] "weight"
#> 
#> attr(,"class")
#> [1] "mlx_param"
#> 
#> [[2]]
#> $env
#> <environment: 0x148637b88>
#> 
#> $name
#> [1] "bias"
#> 
#> attr(,"class")
#> [1] "mlx_param"
#>