Open biochem-fan opened 7 years ago
I get similar unexpected behavior in skcuda.misc.var
.
An axis=0
reshape workaround is also effective for skcuda.misc.var
, which conveniently allows for easy reduction of the GPU memory burden required for var
by splitting the reshaped array into chunks on-GPU, computing var
, downloading to the host, and then reshaping in host RAM once all var
calls are complete.
Hi, it seems that
skcuda.misc.sum
does not support arrays with more than two dimensions.For example, the following code does not work as expected.
For the special case of
axis=0
, a workaround is to reshape the array before summing.I would appreciate a lot if this operation could be supported by the library.