pytorch的backward参数

首先,如果out.backward()中的out是一个标量的话(相当于一个神经网络有一个样本,这个样本有两个属性,神经网络有一个输出)那么此时我的backward函数是不需要输入任何参数的。      运行结果:     不难看出,我们构建了这样的一个函数:       所以其求导也很容易看出:      这是对其进行标量自动求导的结果   如果out.backward()中的out是一个向量(或
相关文章
相关标签/搜索