mechanoChemML.src.transform_layer

Module Contents

Classes

Transform Class to define a Keras transformation layer
class mechanoChemML.src.transform_layer.Transform(transforms, **kwargs)[source]

Bases: tensorflow.keras.layers.Layer

Class to define a Keras transformation layer

Parameters:transforms (func) – A function that takes the input array x and returns a list of transformation outputs
call(self, inputs)[source]
compute_output_shape(self, input_shape)[source]
get_config(self)[source]
classmethod from_config(cls, config)[source]