inst/inst__nn_modules__general/nn_module__nn_unsqueeze.R

In adsb85/lqp: Learning Quadratic Programs

nn_unsqueeze <- torch::nn_module(
  classname = "nn_unsqueeze",
  # --- init
  initialize = function(dim = 2,
                        ...) {
    # --- spec
    self$dim = dim

  },
  # --- forward:
  forward = function(x,...) {

    # --- perform unsqueeze
    x = x$unsqueeze(self$dim)
    return(x)

  }
)