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