标签 "pytorch" 下的文章

基础概念与张量操作1. PyTorch中的Tensor与NumPy数组的主要区别是什么？# 主要区别： # 1. Tensor可以在GPU上运行，NumPy数组只能在CPU上 # 2. Tensor支持自动求导（autograd） # 3. Tensor有更丰富的API用于深度学习 import torch import numpy as np # 相互转换 numpy_array = np.ar...

模块/主题核心类/函数/操作主要用途/说明Tensor操作torch.tensor(), torch.randn(), view(), reshape(), to(device)创建张量、改变形状、设备转换自动微分requires_grad=True, backward(), grad, torch.no_grad()设置梯度追踪、计算梯度、清空梯度、禁用梯度计算数据加载与处理Dataset, D...