Would it be equivelent to the below?
with torch.no_grad():
    preds_array  = preds.cpu().numpy()

Also is there a downside to doing the following instead:
with torch.no_grad():
    outputs  = np_fun(preds.cpu().numpy())