pytorch
Nancyhan88
这个作者很懒,什么都没留下…
展开
-
pytorch 中的DataLoader
DataLoader的作用:通常在训练时我们会将数据集分成若干小的、随机的batch,这个操作当然可以手动操作,但是PyTorch里面为我们提供了API让我们方便地从dataset中获得batch,DataLoader就是干这事儿的。它的本质是一个可迭代对象,一般的操作是:创建一个dataset对象创建一个DataLoader对象遍历这个DataLoader对象,将data, label加载到模型中进行训练#一个粗略的示意dataset = torchvision.datasets.MNIST原创 2020-12-11 10:35:26 · 682 阅读 · 0 评论 -
torch.norm
torch.norm(input, p=‘fro’, dim=None, keepdim=False, out=None, dtype=None)返回所给tensor的矩阵范数或向量范数参数:input:输入tensorp (int, float, inf, -inf, ‘fro’, ‘nuc’, optional):范数计算中的幂指数值。默认为’fro’《pytorch求范数函数——torch.norm》dim (int,2-tuple,2-list, optional): 指定计算的维度。如原创 2020-12-09 10:07:35 · 434 阅读 · 0 评论