Fixed notation issues in joint_tf_wavelet_layer_1d