megnet.layers.readout.linear module¶
Linear readout layer includes stats calculated on the atom dimension
-
class
LinearWithIndex
(*args, **kwargs)[source]¶ Bases:
tensorflow.python.keras.engine.base_layer.Layer
Sum or average the node/edge attributes to get a structure-level vector
- Parameters
mode – (str) ‘mean’, ‘sum’, ‘max’, ‘mean’ or ‘prod’
**kwargs –
-
build
(input_shape)[source]¶ Build tensors :param input_shape: input shapes :type input_shape: sequence of tuple
-
call
(inputs, mask=None)[source]¶ Main logic :param inputs: input tensors :type inputs: tuple of tensor :param mask: mask tensor :type mask: tensor
Returns: output tensor