pytorch中tensor和numpy互相轉換

1. numpy轉化爲tensor.

pytorch 基本函數中的 dim【詳細說明】：以torch.argmax爲例

2. tensor轉化爲numpy

model_corr = model_corr.numpy()

參考鏈接：
https://blog.csdn.net/qianbin3200896/article/details/90233801?depth_1-utm_source=distribute.pc_relevant.none-task&utm_source=distribute.pc_relevant.none-task