How do I perform an average of 2 or more layers i.e. z = keras.layers.average([x, y, w]) as done in Keras. Should I first add them via torch.add and then divide by the number of layers or is there any direct way to perform it?