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