torch.nn系列函数学习
关注
分享
扫一扫
文章平均质量分 67
我不是程序员
这个作者很懒，什么都没留下…
展开
-
pytorch ----【输入张量.data.size()/输入张量.size()】的使用
在提供的代码中，test_data是一个形状为的随机张量，表示一个批次（batch）中有 1 张图像，每张图像有 12 个通道，图像的高度和宽度均为 32 像素。原创 2024-09-24 10:56:27 · 365 阅读 · 0 评论 -
parameters()函数 --- 获取模型参数量
函数是 PyTorch 中类的一个方法，用于返回模型中所有可训练的参数。原创 2024-09-23 23:42:25 · 728 阅读 · 0 评论 -
torch.nn系列函数学习 --- Conv2d函数
是一个正整数时，这种操作也被称为“深度卷积（depthwise convolution）”。，或者是一个整数 / 整数组成的元组，表示在两侧应用的隐式填充量。控制卷积操作的步幅，可以是一个单一的数字或一个元组。控制卷积核点之间的间距，也称为。这个概念较难描述，不过此。控制输入与输出之间的连接。原创 2024-09-23 18:29:38 · 1418 阅读 · 0 评论