Skip to content

Commit 7a57f3a

Browse files
authored
Fix ones_like on GPUs (#142)
1 parent a09f5ee commit 7a57f3a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/utils.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ julia> ones_like(x)
7272
1.0 1.0 1.0 1.0
7373
```
7474
"""
75-
ones_like(x::AbstractArray) = ones(eltype(x), size(x))
75+
ones_like(x::AbstractArray) = fill!(similar(x), 1)
7676
ones_like(x::Number) = oneunit(x)
7777

7878
keep_positive(x::Number) = ifelse(x < 0, zero(x), x) # equivalent to relu

0 commit comments

Comments
 (0)