tensorflow: add Activation, GlobalAveragePooling2D and MaxPool2D layers (#15302)

This commit is contained in:
Hoël Bagard
2026-01-19 17:23:31 +09:00
committed by GitHub
parent d2154f5ad8
commit 64f965e5aa
@@ -443,4 +443,48 @@ class GaussianDropout(Layer[tf.Tensor, tf.Tensor]):
name: str | None = None,
) -> None: ...
class Activation(Layer[tf.Tensor, tf.Tensor]):
def __init__(
self,
activation: _Activation = None,
*,
# **kwargs passed to Layer
# **kwargs passed to Layer
activity_regularizer: _Regularizer = None,
trainable: bool = True,
dtype: _LayerDtype | None = None,
autocast: bool = True,
name: str | None = None,
) -> None: ...
class GlobalAveragePooling2D(Layer[tf.Tensor, tf.Tensor]):
def __init__(
self,
data_format: Literal["channels_last", "channels_first"] | None = None,
keepdims: bool = False,
*,
# **kwargs passed to Layer
activity_regularizer: _Regularizer = None,
trainable: bool = True,
dtype: _LayerDtype | None = None,
autocast: bool = True,
name: str | None = None,
) -> None: ...
class MaxPool2D(Layer[tf.Tensor, tf.Tensor]):
def __init__(
self,
pool_size: int | tuple[int, int] = (2, 2),
strides: int | tuple[int, int] | None = None,
padding: Literal["valid", "same"] = "valid",
data_format: Literal["channels_last", "channels_first"] | None = None,
*,
# **kwargs passed to Layer
activity_regularizer: _Regularizer = None,
trainable: bool = True,
dtype: _LayerDtype | None = None,
autocast: bool = True,
name: str | None = None,
) -> None: ...
def __getattr__(name: str): ... # incomplete module