experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #35871 from Joseph-Rance:patch-3

Browse Source 
PiperOrigin-RevId: 290844721
Change-Id: Ieda686900c4771fa90dd666c5a0b000e2415dedf
This commit is contained in:
TensorFlower Gardener 2020-01-21 16:11:12 -08:00
commit 470239ee94
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tensorflow/python/keras/layers/pooling.py
View File

@ -371,6 +371,24 @@ class MaxPooling2D(Pooling2D):
[11.], [11.],
[12.]]]], dtype=float32)> [12.]]]], dtype=float32)>
Usage Example:
>>> input_image = tf.constant([[[[1.], [1.], [2.], [4.]],
... [[2.], [2.], [3.], [2.]],
... [[4.], [1.], [1.], [1.]],
... [[2.], [2.], [1.], [4.]]]])
>>> output = tf.constant([[[[1], [0]],
... [[0], [1]]]])
>>> model = tf.keras.models.Sequential()
>>> model.add(tf.keras.layers.MaxPooling2D(pool_size=(2, 2),
... input_shape=(4,4,1)))
>>> model.compile('adam', 'mean_squared_error')
>>> model.predict(input_image, steps=1)
array([[[[2.],
[4.]],
[[4.],
[4.]]]], dtype=float32)
For example, for stride=(1,1) and padding="same": For example, for stride=(1,1) and padding="same":
>>> x = tf.constant([[1., 2., 3.], >>> x = tf.constant([[1., 2., 3.],