Browse Source

Update Dense.cs

pull/1094/head
Beacontownfc GitHub 2 years ago
parent
commit
abe33a3829
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      src/TensorFlowNET.Keras/Layers/Core/Dense.cs

+ 2
- 0
src/TensorFlowNET.Keras/Layers/Core/Dense.cs View File

@@ -56,6 +56,7 @@ namespace Tensorflow.Keras.Layers
"kernel",
shape: new Shape(last_dim, args.Units),
initializer: args.KernelInitializer,
regularizer: args.KernelRegularizer,
dtype: DType,
trainable: true);
if (args.UseBias)
@@ -63,6 +64,7 @@ namespace Tensorflow.Keras.Layers
"bias",
shape: new Shape(args.Units),
initializer: args.BiasInitializer,
regularizer: args.BiasRegularizer,
dtype: DType,
trainable: true);



Loading…
Cancel
Save