摘要: 关于自动求导的理解 注意: 调用 backward() 的必须是标量, 若非标量,可先点乘同结构的单位张量,相当于执行一次sum(),从而变成标量。 backward.py import torch from torch.autograd import Variable x = Variable(t 阅读全文
posted @ 2022-02-16 19:11 会飞的斧头 阅读(394) 评论(0) 推荐(0) 编辑
摘要: unsqueeze在指定位置处插入尺寸为1的维度。 注意:返回的张量与此张量共享相同的基础数据。 >>> x = torch.tensor([1, 2, 3, 4]) >>> torch.unsqueeze(x, 0) tensor([[ 1, 2, 3, 4]]) >>> torch.unsque 阅读全文
posted @ 2022-02-16 18:32 会飞的斧头 阅读(2028) 评论(0) 推荐(0) 编辑
摘要: 深度神经网络框架 Pytorch官网安装方法:https://pytorch.org/get-started/locally/ 测试代码 test.py import torch x = torch.rand(5, 3) print("x =", x) print("x.shape =",x.sha 阅读全文
posted @ 2022-02-16 17:08 会飞的斧头 阅读(311) 评论(0) 推荐(0) 编辑
摘要: 转自: https://blog.csdn.net/perfect2011/article/details/120255629 pytorch 和tensorflow 中最重要的概念就是tensor了,tensorflow 这个框架的名字中很直白,就是tensor的流动, 所以学习深度学习的第一课就 阅读全文
posted @ 2022-02-16 17:05 会飞的斧头 阅读(1429) 评论(0) 推荐(0) 编辑