causalTest.py

import tensorflow as tf
from tensorflow.keras.layers import Conv1D


class CausalConv(tf.keras.Model):
    def __init__(self):
        super(CausalConv, self).__init__()
        self.conv = Conv1D(filters=16, kernel_size=2, padding='causal')

    def call(self, x):
        return self.conv(x)

x = tf.zeros([1, 128, 1])
model = CausalConv()
y = model(x)
print('input shape:', x.shape)
print('output shape:', y.shape)