View source on GitHub |
Cropping layer for 1D input (e.g. temporal sequence).
tf.keras.layers.Cropping1D( cropping=(1, 1), **kwargs )
It crops along the time dimension (axis 1).
input_shape = (2, 3, 2) x = np.arange(np.prod(input_shape)).reshape(input_shape) print(x) [[[ 0 1] [ 2 3] [ 4 5]] [[ 6 7] [ 8 9] [10 11]]] y = tf.keras.layers.Cropping1D(cropping=1)(x) print(y) tf.Tensor( [[[2 3]] [[8 9]]], shape=(2, 1, 2), dtype=int64)
Arguments | |
---|---|
cropping | Int or tuple of int (length 2) How many units should be trimmed off at the beginning and end of the cropping dimension (axis 1). If a single int is provided, the same value will be used for both. |
3D tensor with shape (batch_size, axis_to_crop, features)
3D tensor with shape (batch_size, cropped_axis, features)
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/keras/layers/Cropping1D