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