Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update pooling.py
Add Integer in `strides` and `pool_size` of 3D layers
Added "If None, it will default to `pool_size`." to be consistent with explanation of 1D, 2D layer
  • Loading branch information
Naruu authored Oct 19, 2019
commit f23a548544016f7b70935967b16abca01de37ad3
8 changes: 5 additions & 3 deletions keras/layers/pooling.py
Original file line number Diff line number Diff line change
Expand Up @@ -390,7 +390,8 @@ class MaxPooling3D(_Pooling3D):
pool_size: Integer or tuple of 3 integers,
factors by which to downscale (dim1, dim2, dim3).
(2, 2, 2) will halve the size of the 3D input in each dimension.
strides: tuple of 3 integers, or None. Strides values.
strides: Integer, tuple of 3 integers, or None. Strides values.
If None, it will default to `pool_size`.
padding: One of `"valid"` or `"same"` (case-insensitive).
data_format: A string,
one of `channels_last` (default) or `channels_first`.
Expand Down Expand Up @@ -437,10 +438,11 @@ class AveragePooling3D(_Pooling3D):
"""Average pooling operation for 3D data (spatial or spatio-temporal).

# Arguments
pool_size: tuple of 3 integers,
pool_size: Integer or tuple of 3 integers,
factors by which to downscale (dim1, dim2, dim3).
(2, 2, 2) will halve the size of the 3D input in each dimension.
strides: tuple of 3 integers, or None. Strides values.
strides: Integer, tuple of 3 integers, or None. Strides values.
If None, it will default to `pool_size`.
padding: One of `"valid"` or `"same"` (case-insensitive).
data_format: A string,
one of `channels_last` (default) or `channels_first`.
Expand Down