layer_dense_torch {deepregression} | R Documentation |
Function to define a torch layer similar to a tf dense layer
Description
Function to define a torch layer similar to a tf dense layer
Usage
layer_dense_torch(
input_shape,
units = 1L,
name,
trainable = TRUE,
kernel_initializer = "glorot_uniform",
use_bias = FALSE,
kernel_regularizer = NULL,
...
)
Arguments
input_shape |
integer; number of input units |
units |
integer; number of output units |
name |
string; string defining the layer's name |
trainable |
logical; whether layer is trainable |
kernel_initializer |
initializer; for coefficients |
use_bias |
logical; wether bias is used (default no) |
kernel_regularizer |
regularizer; for coefficients |
... |
arguments used in choose_kernel_initializer_torch |
Value
torch layer
[Package deepregression version 2.2.0 Index]