Skip to content

Commit 00ca29c

Browse files
committed
minor ed in nsf
1 parent aa2adeb commit 00ca29c

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

src/flows/neuralspline.jl

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -8,19 +8,19 @@ Neural Rational quadratic Spline layer
88
[1] Durkan, C., Bekasov, A., Murray, I., & Papamakarios, G., Neural Spline Flows, CoRR, arXiv:1906.04032 [stat.ML], (2019).
99
"""
1010
struct NeuralSplineLayer{T,A<:Flux.Chain} <: Bijectors.Bijector
11-
dim::Int # dimension of input
12-
K::Int # number of knots
13-
n_dims_transferred::Int # number of dimensions that are transformed
14-
nn::A # networks that parmaterize the knots and derivatives
15-
B::T # bound of the knots
11+
dim::Int # dimension of input
12+
K::Int # number of knots
13+
n_dims_transferred::Int # number of dimensions that are transformed
14+
nn::A # networks that parmaterize the knots and derivatives
15+
B::T # bound of the knots
1616
mask::Bijectors.PartitionMask
1717
end
1818

1919
function NeuralSplineLayer(
20-
dim::T1, # dimension of input
21-
hdims::T1, # dimension of hidden units for s and t
22-
K::T1, # number of knots
23-
B::T2, # bound of the knots
20+
dim::T1, # dimension of input
21+
hdims::T1, # dimension of hidden units for s and t
22+
K::T1, # number of knots
23+
B::T2, # bound of the knots
2424
mask_idx::AbstractVector{<:Int}, # index of dimensione that one wants to apply transformations on
2525
) where {T1<:Int,T2<:Real}
2626
num_of_transformed_dims = length(mask_idx)

0 commit comments

Comments
 (0)