看起来我们没有用到ai
import torch nn as nn :
am I a joke to you?
还有ai是拿去跟train data 做比对
这理解也是很前卫