Skip to content

Commit 1dba1a6

Browse files
Update test/layers/basic.jl
Co-authored-by: Dhairya Gandhi <dhairya@juliacomputing.com>
1 parent 2de5745 commit 1dba1a6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/layers/basic.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,7 @@ import Flux: activations
210210
x = rand(1:vocab_size, 3, 4)
211211
y = m(x)
212212
@test y isa Array{Float32, 3}
213-
@test size(y) == (embed_size, 3, 4)
213+
@test size(y) == (embed_size, 3, 4)
214214

215215
@test m(2) m.weight[:,2]
216216
@test m(OneHotVector(3, vocab_size)) m.weight[:,3]

0 commit comments

Comments
 (0)