![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习框架
记录学习AI框架
USTC_SC
这个作者很懒,什么都没留下…
展开
-
Pytorch学习笔记(2)-常用函数与概念
文章目录1.指定运行设备2.稀疏矩阵3.四则运算4.矩阵运算5.幂、开方、对数运算6.in-place操作7.广播机制8.取整、取余、比较、top-k9.判定异常值10.三角函数11.统计函数 1.指定运行设备 pytorch可以指定数据运行在cpu或者gpu上,以下为例。 import torch #dev = torch.device("cpu") dev = torch.device("cuda") a = torch.tensor([2, 2], dtype=to原创 2020-08-16 20:53:21 · 721 阅读 · 0 评论 -
pytorch学习笔记(1)-从创建Tensor开始
文章目录创建Tensor1.基本创建2.创建特殊值3.创建随机值和特定序列Tensor4.Numpy转Tensor5.Tensor转Numpy 创建Tensor pytorch学习的最基础的学习就从创建张量开始。 1.基本创建 import torch #根据值创建 a = torch.Tensor([[1, 2],[3, 4]]) print(a) #根据形状创建,随机初始 b = torch.Tensor(2, 2) print(b) d = torch.tensor(((1, 2), (3, 4原创 2020-08-16 20:40:13 · 177 阅读 · 0 评论