layer_dense_module: Function to create custom nn_linear module to overwrite...

View source: R/layers_torch.R

layer_dense_moduleR Documentation

Function to create custom nn_linear module to overwrite reset_parameters

Description

Function to create custom nn_linear module to overwrite reset_parameters

Usage

layer_dense_module(kernel_initializer)

Arguments

kernel_initializer

string; initializer used to reset_parameters

Value

nn module


deepregression documentation built on Sept. 9, 2025, 5:27 p.m.