Fixed notations in joint_tf_wavelet_layer_1d