Skip to content

Commit fe09a9d

Browse files
committed
add tf.keras.layers.MaxPool2D
1 parent 368b8d3 commit fe09a9d

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

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

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -471,4 +471,20 @@ class GlobalAveragePooling2D(Layer[tf.Tensor, tf.Tensor]):
471471
name: str | None = None,
472472
) -> None: ...
473473

474+
class MaxPool2D(Layer[tf.Tensor, tf.Tensor]):
475+
def __init__(
476+
self,
477+
pool_size: int | tuple[int, int] = (2, 2),
478+
strides: int | tuple[int, int] | None = None,
479+
padding: Literal["valid", "same"] = "valid",
480+
data_format: Literal["channels_last", "channels_first"] | None = None,
481+
*,
482+
# **kwargs passed to Layer
483+
activity_regularizer: _Regularizer = None,
484+
trainable: bool = True,
485+
dtype: _LayerDtype | None = None,
486+
autocast: bool = True,
487+
name: str | None = None,
488+
) -> None: ...
489+
474490
def __getattr__(name: str): ... # incomplete module

0 commit comments

Comments
 (0)