Well you can either use a layer and fix the parameter or use the functional interface where you pass the weights. I did a small demo of a discrete wavelet transform using the latter 11. Maybe you can take bits of that for your inspiration.
Best regards
Thomas