# ----------- 判断模型是在CPU还是GPU上 ----------------------model = nn.LSTM(input_size=10, hidden_size=4, num_layers=1, batch_first=True)print(next(model.parameters()).device)# 输出：cpumodel = model.cuda()print