PyTorch接口总结1
需积分: 0 137 浏览量
2022-08-03
17:31:49
上传
评论
收藏 507KB PDF 举报
Data Type CPU tensor GPU tensor
32-bit floating point torch.FloatTensor torch.cuda.FloatTensor
64-bit floating point torch.DoubleTensor torch.cuda.DoubleTensor
16-bit floating point torch.HalfTensor torch.cuda.HalfTensor
8-bit integer
(unsigned)
torch.ByteTensor torch.cuda.ByteTensor
8-bit integer (signed) torch.CharTensor torch.cuda.CharTensor
16-bit integer (signed) torch.ShortTensor torch.cuda.ShortTensor
32-bit integer (signed) torch.IntTensor torch.cuda.IntTensor
PyTorch接口总结
1.Tensor
1.1 Tensor类型
通常使用 Tensor 类的构造函数返回的是 FloatTensor 类型对象，可以通过在对象
上调用 cuda() 返回一个新的 cuda.FloatTensor 类型的对象。
torch 模块内提供了操作tensor的接口，而 Tensor 类型的对象上也设计了对应的接
口。例如： torch.add() 与 tensor.add() 等价。
想就地修改一个tensor对象，需要使用加后缀下划线的方法。
剩余12页未读，继续阅读
伯特兰·罗卜
- 粉丝: 21
- 资源: 309
最新资源
- 基于matlab实现电力系统仿真计算软件包，包括潮流计算，最优潮流计算等.rar
- 基于matlab实现电力系统各种故障波形仿真，单相接地故障，两相间短路，两相接地短路，三相短路等.rar
- 基于matlab实现电动汽车动力性，爬坡性，续驶里程等性能仿真.rar
- Python动态烟花代码.pdf
- 基于matlab实现串口发送接收数据 可配置端口，波特率等 发送可选择ASCII方式或HEX方式
- matlab基于BP神经网络手写字母识别（单一）.zip代码9
- 基于matlab实现编写的串口调试工具，数据接收部分采用中断方式，保证了实时的数据显示
- 基于matlab实现39节点电力系统合闸角调控过程中的机组和负荷的灵敏度计算.rar
- HBase数据库性能调优
- 原生微信小程序源码 - -首字母排序选择
资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！
点击此处反馈
评论0