Add tests

findmyway · Oct 23, 2019 · f7ce717 · f7ce717
1 parent e0c1c0e
commit f7ce717
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/cuda/cuda.jl b/test/cuda/cuda.jl
@@ -28,6 +28,8 @@ cm = gpu(m)
 x = [1,2,3]
 cx = gpu(x)
 @test Flux.crossentropy(x,x) ≈ Flux.crossentropy(cx,cx)
+@test Flux.crossentropy(x,x, weight=1.0) ≈ Flux.crossentropy(cx,cx, weight=1.0)
+@test_broken Flux.crossentropy(x,x, weight=[1.0;2.0;3.0]) ≈ Flux.crossentropy(cx,cx, weight=[1.0;2.0;3.0])
 
 xs = rand(5, 5)
 ys = Flux.onehotbatch(1:5,1:5)