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 Jan. 24, 2023, 1:05 a.m.