clone_module function

Clone a torch module.

Clones a module.

clone_module(module, deep = FALSE, ..., replace_values = TRUE)

Arguments

• module: (nn_module)

  The module to clone

• deep: (logical(1))

  Whether to create a deep clone.

• ...: (any)

  Additional parameters, currently unused.

• replace_values: (logical(1))

  Whether to replace parameters and buffers with the cloned values.

Examples

if (torch_is_installed()) {
clone_module(nn_linear(1, 1), deep = TRUE)
# is the same as
nn_linear(1, 1)$clone(deep = TRUE)
}