- 博客(3)
- 收藏
- 关注
RSS订阅原创 pytorch -- 张量相关方法
本文主要讲解ndimension()、nelement()、size() / shape、view() / reshape等与张量相关的方法 深度学习中，张量使用相当频繁，本篇文章主要讲解一些与张量有关的方法。萌新博主，还请各位大佬指导o(*￣▽￣*)ブ 读者如果要尝试本篇中的代码，请自行下载pytorch框架，并且引入torch和numpy包。 先生成示例张量 t （本文后续操作会在 t张量之上进行） # 张量相关方法demo import torch import numpy as np .
2021-01-05 23:39:21
551
原创 pytorch -- 张量的存储设备
张量的存储设备可以是CPU或者GPU，在没有特别说明的情况下，默认存储在CPU中，我们可以在创建张量的过程中，指明张量存储的设备 存储在CPU中 存储在GPU中（由于我的笔记本电脑无独显，并且采用的是锐龙处理器，所以无法存储在GPU中，读者可以根据自己的电脑自行尝试） ...
2021-01-04 15:50:55
381
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人