forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathILayer.cs
More file actions
32 lines (31 loc) · 1.19 KB
/
ILayer.cs
File metadata and controls
32 lines (31 loc) · 1.19 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
using Tensorflow.Common.Types;
using Tensorflow.Keras.Engine;
using Tensorflow.Keras.Saving;
using Tensorflow.NumPy;
using Tensorflow.Training;
namespace Tensorflow.Keras
{
public interface ILayer: IWithTrackable, IKerasConfigable
{
string Name { get; }
bool Trainable { get; }
bool Built { get; }
void build(KerasShapesWrapper input_shape);
List<ILayer> Layers { get; }
List<INode> InboundNodes { get; }
List<INode> OutboundNodes { get; }
Tensors Apply(Tensors inputs, Tensors states = null, bool? training = false, IOptionalArgs? optional_args = null);
List<IVariableV1> TrainableVariables { get; }
List<IVariableV1> TrainableWeights { get; }
List<IVariableV1> NonTrainableWeights { get; }
List<IVariableV1> Weights { get; set; }
void set_weights(IEnumerable<NDArray> weights);
List<NDArray> get_weights();
Shape OutputShape { get; }
KerasShapesWrapper BatchInputShape { get; }
KerasShapesWrapper BuildInputShape { get; }
TF_DataType DType { get; }
int count_params();
void adapt(Tensor data, int? batch_size = null, int? steps = null);
}
}