u012992177的博客

原创 pytorch笔记-batch

在学习莫烦大神的pytorch视频的batch部分，由于pytorch版本更新，产生了一些不兼容的情况。源代码如下：import torchimport torch.utils.data as Datatorch.manual_seed(1) # 设定随机数种子BATCH_SIZE=5x=torch.linspace(1,10,10)y=torch.linspace(10,1,10)...