deepregression (version 2.2.0)

layer_dense_torch: 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,
  ...
)

Value

torch layer

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