摘要: Tensor变化 导入库 import numpy as np # 导入Numpy import torch # 导入Torch 1. 创建张量 1)通过numpy的array创建 # 创建(3,3)的全1矩阵 arr = np.ones((3, 3)) # 创建张量,默认存放在CPU t1 = t 阅读全文
posted @ 2021-06-09 14:12 Haozi_D17 阅读(99) 评论(0) 推荐(0) 编辑
摘要: GPU加速 1. 定义GPU设备 import torch device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") print(device) 2. 将模型、张量等放在GPU设备上 # loss.to(devic 阅读全文
posted @ 2021-06-09 13:10 Haozi_D17 阅读(53) 评论(0) 推荐(0) 编辑
摘要: 通用结构 首先:导入相关库 import numpy as np import pandas as pd import matplotlib.pyplot as plt import torch from torch import nn from torch.autograd import Vari 阅读全文
posted @ 2021-06-09 12:54 Haozi_D17 阅读(62) 评论(0) 推荐(0) 编辑