terior_fn <span class='op'>=</span> <span class='va'>tfp</span><span class='op'>$</span><span class='va'>layers</span><span class='op'>$</span><span class='va'>util</span><span class='op'>$</span><span class='fu'>default_mean_field_normal_fn</span><span class='op'>(</span><span class='op'>)</span>,
  kernel_posterior_tensor_fn <span class='op'>=</span> <span class='kw'>function</span><span class='op'>(</span><span class='va'>d</span><span class='op'>)</span> <span class='va'>d</span> <span class='op'>%&gt;%