Make extract_columns work with 3D arrays
Bugfix in column.py
Aggregate mini-batch losses on CPU