api:
參數:
默認加載方式:使用cpu加載cpu訓練的模型或使用gpu調用gpu訓練的模型;
在cpu上加載所有張量:
使用函數將所有張量加載到CPU中(適合GPU訓練的模型加載到CPU中):
將所有張量加載到第壹個GPU中(在CPU中訓練並在GPU中加載):
從GPU 1到GPU 0的映射張量(第壹次GPU訓練,第二次GPU加載):
根據您的設備,將張量加載到您的當前設備中: