.detach().cpu().numpy()

Picassooo / 2024-10-23 / 原文

转:PyTorch关于以下方法使用:detach() cpu() numpy() 以及item()_output.cpu().detach()-CSDN博客