Gpu diagonal subtensor

@jaberg, can you review this rapidly? We need this for a deadline.