sharanry

sharanry

sharanry

sharanry

using Flux, Flux.Tracker
using Flux.Tracker: grad, update!
using Distributions
using StatsFuns
using MLToolkit
using LinearAlgebra
using Random

abstract type Flow end

sharanry

sharanry

sharanry

using Flux, Flux.Tracker
using Flux.Tracker: grad, update!
using LinearAlgebra
using Distributions

function functional(i)
    u, w, b = uₖ[i], wₖ[i],bₖ[i]
    f(z) = z + u*tanh.(transpose(w)*z + b)
end
dtanh(z) = 1-tanh.(z).^2

sharanry

sharanry

sharanry