We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0773796 commit d219cdbCopy full SHA for d219cdb
test/activation.jl
@@ -100,7 +100,7 @@ end
100
xs = Float32[1 2 3; 1000 2000 3000]
101
@test logsoftmax(xs) ≈ [-999 -1998 -2997; 0 0 0.]
102
103
- @test NNlib.∇logsoftmax(ones(size(xs)), xs) ≈ zeros(Float32, size(xs))
+ @test NNlib.∇logsoftmax(ones(size(xs)), xs) ≈ Float32[1 1 1; -1 -1 -1]
104
@test NNlib.∇softmax(ones(size(xs)), xs) ≈ zeros(Float32, size(xs))
105
106
# These values precalculated using PyTorch's nn.LogSoftmax
0 commit comments