Skip to content

Commit 368b8d3

Browse files
committed
add tf.keras.layers.GlobalAveragePooling2D
1 parent 494a4b3 commit 368b8d3

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

stubs/tensorflow/tensorflow/keras/layers/__init__.pyi

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -457,4 +457,18 @@ class Activation(Layer[tf.Tensor, tf.Tensor]):
457457
name: str | None = None,
458458
) -> None: ...
459459

460+
class GlobalAveragePooling2D(Layer[tf.Tensor, tf.Tensor]):
461+
def __init__(
462+
self,
463+
data_format: Literal["channels_last", "channels_first"] | None = None,
464+
keepdims: bool = False,
465+
*,
466+
# **kwargs passed to Layer
467+
activity_regularizer: _Regularizer = None,
468+
trainable: bool = True,
469+
dtype: _LayerDtype | None = None,
470+
autocast: bool = True,
471+
name: str | None = None,
472+
) -> None: ...
473+
460474
def __getattr__(name: str): ... # incomplete module

0 commit comments

Comments
 (0)