We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8e805ef commit 033daf2Copy full SHA for 033daf2
src/transform/transform.jl
@@ -5,6 +5,9 @@ Abstract type defining a transformation of the input.
5
"""
6
abstract type Transform end
7
8
+# We introduce our own _map for Transform so that we can work around
9
+# https://github.com/FluxML/Zygote.jl/issues/646 and define our own pullback
10
+# (see zygoterules.jl)
11
Base.map(t::Transform, x::AbstractVector) = _map(t, x)
12
_map(t::Transform, x::AbstractVector) = t.(x)
13
0 commit comments