deephyper.keras.layers.GlobalAvgPool#

class deephyper.keras.layers.GlobalAvgPool(*args: Any, **kwargs: Any)[source]#

Bases: Layer

Global Average Pool.

Takes the average over all the nodes or features. Details can be seen from danielegrattarola/spektral

Parameters:

axis (int) – the axis to take average.

Methods

call

Apply the layer on input tensors.

__call__(*args: Any, **kwargs: Any) Any#

Call self as a function.

call(inputs, **kwargs)[source]#

Apply the layer on input tensors.

Parameters:

inputs (tensor) – the node feature tensor

Returns:

GlobalAvgPool tensor (tensor)