Tensorflow.js tf.layers.average() Function
The tf.layers.average() function is used to calculate average of every element when multiple arrays are provided as inputs.
tf.layers.average (inputShape?, batchInputShape?, batchSize?, dtype?, name?, trainable?, weights?, inputDType? )
- inputShape : It is an optional parameter which is used to create the input layer and it take values like number and null.
- batchInputShape : It is an optional parameter which is used to create the input layer before the main layer and it take values like number and null.
- batchSize : Its is an optional parameter used to make batchInputShape and it accepts only numbers.
- dtype : It is an optional parameter and it stands for data type. By default it has ‘float32’ and also supports other values like ‘int32’, ‘bool’ etc.
- name: It is an optional parameter and is used to define the name of the layer and it accepts strings.
- trainable : It is an optional parameter that determines the provided input layers are updated or not. It accepts boolean values.
- weights : It possesses the starting weights of the layer. It is also a optional parameter.
- inputDType : It is an optional parameter used for input data type. Like dtype it also supports all its values.
Return Value: It returns average.
[null, 33, 3]
[null, 4, 4]