Test even more dimensions

Co-authored-by: Michael Abbott <32575566+mcabbott@users.noreply.github.com>
FluxML · Jun 27, 2021 · 2c1f05c · 2c1f05c
1 parent 7f6d899
commit 2c1f05c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/softmax.jl b/test/softmax.jl
@@ -1,5 +1,5 @@
 @testset "softmax" begin
-    for (sz, dims) in [((5,), :), ((5,), 1), ((5,5), :), ((5,5), 1), ((5,5), 2)]
+    for (sz, dims) in [((5,), :), ((5,), 1), ((5,5), :), ((5,5), 1), ((5,5), 2), ((5,5,5,5), (2,3)), ((5,5,5,5), (2,4))]
         x = randn(Float64, sz)
         y = softmax(x, dims=dims)
         dy = randn(Float64, sz)