Skip to contents

Creates an nn_parameter

Source: R/nn.R
nn_parameter.Rd

Indicates to nn_module that x is a parameter

Usage

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