Should it really be necessary to do var.detach().cpu().numpy()?

SimonW · July 9, 2019, 2:12am

The reason for requiring explicit .cpu() is that CPU tensors and the converted numpy arrays share memory. If a .cpu() is implicitly done, the operation will be different for CUDA and CPU tensors, and we wanted to be explicit to avoid bugs.