Closed andyyankai closed 4 years ago
Also, it seem scatter_add is working, but I hope to know if there is any way to avoid the loop, just for the performance reason
load<float>(ptr)
will return a simple float
as this is the type you specified as template parameter. (here the first one of the list, so 10.f
).
You can construct the FloatC
passing the list-initializion I believe:
FloatC arr_enoki = { 1.f, 2.f, 3.f, 4.f };
You should also give a try to functions like linspace<FloatC>
and arange<FloatC>
, those are really useful.
I tried