Tensorflow ReflectionPad1d module.
| 9 | |
| 10 | |
| 11 | class TFReflectionPad1d(tf.keras.layers.Layer): |
| 12 | """Tensorflow ReflectionPad1d module.""" |
| 13 | |
| 14 | def __init__(self, padding_size): |
| 15 | """Initialize TFReflectionPad1d module. |
| 16 | |
| 17 | Args: |
| 18 | padding_size (int): Padding size. |
| 19 | |
| 20 | """ |
| 21 | super(TFReflectionPad1d, self).__init__() |
| 22 | self.padding_size = padding_size |
| 23 | |
| 24 | @tf.function |
| 25 | def call(self, x): |
| 26 | """Calculate forward propagation. |
| 27 | |
| 28 | Args: |
| 29 | x (Tensor): Input tensor (B, T, 1, C). |
| 30 | |
| 31 | Returns: |
| 32 | Tensor: Padded tensor (B, T + 2 * padding_size, 1, C). |
| 33 | |
| 34 | """ |
| 35 | return tf.pad(x, [[0, 0], [self.padding_size, self.padding_size], [0, 0], [0, 0]], "REFLECT") |
| 36 | |
| 37 | |
| 38 | class TFConvTranspose1d(tf.keras.layers.Layer): |