睿睿睿睿-优快云博客

原创 Pytorch-反向传播后梯度为None

经过这个步骤后可以判断哪些变量没有参与梯度计算，将其从模型中去掉即可。

2023-12-21 19:57:39 1024 2

原创 torch.nn.CosineSimilarity() 三维张量

对于三维张量：import torcha = torch.tensor([[[-1., 0]], [[1., 0]]])b = torch.tensor([[[1., 1.], [2., 1.]]])print(a.shape)print(b.shape)x = torch.nn.CosineSimilarity(-1)result = x(a, b)print(result)输出：torch.Size([2, 1, 2]) # 张量a的形状torch.Size([1,

2022-05-23 18:09:39 3564

空空如也

Tomcat8.0可以通过浏览器用8080窗口访问到，但是为什么eclipse添加服务器报错端口错误？

2020-02-18

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

qq_45189662的博客

原创 Pytorch-反向传播后梯度为None

原创 torch.diag()

原创 torch.nn.CosineSimilarity() 三维张量

空空如也

Tomcat8.0可以通过浏览器用8080窗口访问到，但是为什么eclipse添加服务器报错端口错误？