pytorch中backward()函数的gradient参数作用
摘要:
若是关于计算图中的leaves变量求导的输出变量是一个标量,那么gradient默认为None,或者指定为“torch.Tensor([1.0])” 若是关于计算图中的leaves求导的输出变量是一个向量,那么gradient是不能缺省的,gradient是和该输出向量同维度的tensor。 假设计 阅读全文
posted @ 2021-12-19 22:09 那抹阳光1994 阅读(504) 评论(0) 推荐(0) 编辑