TensorFlow 1 version
|
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).
Examples:
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. |
Input shape:
3D tensor with shape (batch_size, axis_to_crop, features)
Output shape:
3D tensor with shape (batch_size, cropped_axis, features)
TensorFlow 1 version
View source on GitHub