摘要: import torch from torch import nn # view函数的-1参数的作用在于基于另一参数,自动计算该维度的大小 # view的第一个参数:2 代表的是batch 后面的2,2应该是几行几列的意思,这两个必须一致 # arange(1,9) => 1,2,3,4,5,6,7 阅读全文
posted @ 2021-11-25 09:30 我太想努力了 阅读(186) 评论(0) 推荐(0)