search

denizyuret / AutoGrad.jl

Julia port of the Python autograd package.
Other
169 stars 26 forks source link

mapslices and Param wrapped arrays #110

Open hiiroo opened 5 years ago

hiiroo commented 5 years ago

I was trying mapslices function however, I've realized that mapslices function restricted to subtypes of AbstractArray and does not accept Param wrapped arrays.

Here is a line of code to demonstrate the issue. mapslices(x->x*rand(4,2), Param(rand(3,4,5)), dims=[1,2])