Skip to content

Commit a80bea9

Browse files
SomTambemcabbott
andauthored
Add rng which I had forgotten.
Co-authored-by: Michael Abbott <32575566+mcabbott@users.noreply.github.com>
1 parent 54bf710 commit a80bea9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/utils.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -218,8 +218,8 @@ true
218218
[1] Saxe, McClelland, Ganguli. "Exact solutions to the nonlinear dynamics of learning in deep linear neural networks", ICLR 2014, https://arxiv.org/abs/1312.6120
219219
220220
"""
221-
function orthogonal(rng::AbstractRNG, rows, cols; gain = 1)
222-
mat = rows > cols ? randn(Float32, rows, cols) : randn(Float32, cols, rows)
221+
function orthogonal(rng::AbstractRNG, rows::Integer, cols::Integer; gain = 1)
222+
mat = rows > cols ? randn(rng, Float32, rows, cols) : randn(rng, Float32, cols, rows)
223223

224224
Q, R = LinearAlgebra.qr(mat)
225225
Q = Array(Q) * sign.(LinearAlgebra.Diagonal(R))

0 commit comments

Comments
 (0)