Skip to contents

Assign arrays back to parameters

mlx_param_set_values.Rd

Assign arrays back to parameters

Usage

mlx_param_set_values(params, values)

Arguments

params

A list of mlx_param.

values

A list of arrays.

See also

mlx.nn.Module.update

Examples

set.seed(1)
layer <- mlx_linear(2, 1)
params <- mlx_parameters(layer)
current <- mlx_param_values(params)
mlx_param_set_values(params, current)