博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
pytorch 2 variable 变量
阅读量:4687 次
发布时间:2019-06-09

本文共 983 字,大约阅读时间需要 3 分钟。

import torchfrom torch.autograd import Variabletensor = torch.FloatTensor([[1, 2], [3, 4]])variable = Variable(tensor, requires_grad=True)print(tensor)    # tensor([[1., 2.], [3., 4.]])print(variable)  # tensor([[1., 2.], [3., 4.]], requires_grad=True) 梯度设置为True,反向传播要使用t_out = torch.mean(tensor*tensor)  # x^2v_out = torch.mean(variable*variable)print(t_out)      # tensor(7.5000)   变量计算的时候比张量多了一个梯度信息print(v_out)      # tensor(7.5000, grad_fn=
)v_out.backward() # 反向传播# v_out = 1/4*sun(var*var) # 求导过程# d(v_out)/d(var) = 1/4*2*variable = variable/2 # 求导过程print(variable.grad) # tensor([[0.5000, 1.0000], [1.5000, 2.0000]])print("variable:", variable) # tensor([[1., 2.], [3., 4.]], requires_grad=True)print("variable.data:", variable.data) # tensor([[1., 2.], [3., 4.]])print("variable.data.numpy:", variable.data.numpy()) # [[1. 2.] [3. 4.]]

END

转载于:https://www.cnblogs.com/yangzhaonan/p/10439422.html

你可能感兴趣的文章
selenium无法正常运行 Chrome浏览器,cannot find Chrome binary的问题
查看>>
一体机分区误删找到数据的方案
查看>>
excel常用函数
查看>>
网络协议-restful协议
查看>>
JavaScript模块化编程(一)
查看>>
egg文件制作与安装
查看>>
后台测试流程与经验分享
查看>>
EventBus 最简易的使用方式
查看>>
jQuery与Dom
查看>>
统治世界的十大算法
查看>>
Microsoft Office word powerpoint 中删除MathType加载项后每次启动显示加载错误
查看>>
剑指offer54 表示数值的字符串
查看>>
h5py
查看>>
网络的四层架构与网站的数据库的用户信息表的设计
查看>>
响应在此上下文中不可用
查看>>
Mysql入门-基本操作(一)
查看>>
git-gui
查看>>
splay入门教程
查看>>
Queryable.Union 方法实现json格式的字符串合并
查看>>
福大软工1816:Beta总结
查看>>