There are quite a few problems using `af::array` to handle CPU <-> GPU transitions. A swap to ndarray would be cleaner for host data
There are quite a few problems using
af::arrayto handle CPU <-> GPU transitions. A swap to ndarray would be cleaner for host data