nn_parameter: Creates an 'nn_parameter'

View source: R/nn.R

nn_parameterR Documentation

Creates an nn_parameter

Description

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


torch documentation built on June 7, 2023, 6:19 p.m.