Skip to content

Commit a3e6457

Browse files
committed
Fix dropdims deprecation.
1 parent 5b64ed9 commit a3e6457

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/conv.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ end
8989
W = copy(permutedims(w[:,:,:,i:i],[1,2,4,3]));
9090
DY = copy(dy[:,:,2i-1:2i,:]);
9191
res = ∇conv_data(DY,X,W)
92-
@test dropdims(z[:,:,i:i,:], (3,4)) == dropdims(res, (3,4))
92+
@test dropdims(z[:,:,i:i,:], dims=(3,4)) == dropdims(res, dims=(3,4))
9393
end
9494

9595
z = ∇depthwiseconv_filter(dy, x, w)

0 commit comments

Comments
 (0)