摘要: 博客地址:https://www.cnblogs.com/zylyehuo/ 一、导入所用库 import torch from torch import nn from d2l import torch as d2l 二、自定义重塑层 class Reshape(nn.Module): def f 阅读全文
posted @ 2025-07-11 21:28 zylyehuo 阅读(240) 评论(0) 推荐(0)
摘要: 博客地址:https://www.cnblogs.com/zylyehuo/ 基本设置 import torch from torch import nn from d2l import torch as d2l batch_size = 256 train_iter, test_iter = d2 阅读全文
posted @ 2025-07-11 21:08 zylyehuo 阅读(32) 评论(0) 推荐(0)
摘要: 博客地址:https://www.cnblogs.com/zylyehuo/ 一. 数据准备阶段 1. 导入库 %matplotlib inline import random import matplotlib.pyplot as plt import torch from d2l import 阅读全文
posted @ 2025-07-11 14:55 zylyehuo 阅读(17) 评论(0) 推荐(0)