随笔分类 -  PyTorch

摘要:先看看文档怎么说的 这里说指定某个纬度(dim),就会让那个纬度的和为1 看一下input是2维的情况 dim=0 dim=1 再看看input是3维的情况 dim = 0 dim = 1 阅读全文
posted @ 2022-04-04 00:03 裏表異体 阅读(39) 评论(0) 推荐(0) 编辑
摘要:Pytorch简介 PyTorch是一个基于Torch的Python开源机器学习库,用于自然语言处理等应用程序。它主要由Facebookd的人工智能小组开发,不仅能够 实现强大的GPU加速,同时还支持动态神经网络,这一点是现在很多主流框架如TensorFlow都不支持的。 PyTorch提供了两个高 阅读全文
posted @ 2022-02-22 19:44 裏表異体 阅读(1701) 评论(0) 推荐(0) 编辑
摘要:环境: Python版本: 3.8 IDE:Spyder 5.2.2 代码: from torchvision import datasets from matplotlib import pyplot as plt # 下载数据 from torchvision import datasets f 阅读全文
posted @ 2022-02-21 14:44 裏表異体 阅读(116) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示