Module's cannot contain other module classes

[sebffischer]

library(torch)
nn_module2 = nn_module("nn_module2",
  initialize = function() {
    self$layer = self$nn_layer()
  },
  nn_layer = nn_module("layer",
    initialize = function() {
      self$linear = nn_linear(10, 1)
    },
    forward = function(x) {
      self$linear(x)
    }
  ),
  forward = function(x) {
    self$layer(x)
  }
)

nn_module2()
#> Error in self$nn_layer(): object 'Module' not found

^{Created on 2025-05-28 with reprex v2.1.1}