Indicates to nn_module that x is a parameter

nn_parameter(x, requires_grad = TRUE)

Arguments

x

the tensor that you want to indicate as parameter

requires_grad

whether this parameter should have requires_grad = TRUE