随笔分类 - PyTorch
摘要：实际上无论是几维，方法都是一样。假设以 torch.softmax() 为例： # 下面运行结果所使用的代码 import torch import numpy as np z = np.arange(1, 33).reshape((2, 2, 2, 4)) z = torch.tensor(z,
阅读全文
摘要：自己学深度学习的时候想要记下来的，看到代码这样用就这样记了，所以我也不知道这个定义过程规不规范，反正就是能用。其他类型的数据按代码改一下应该没问题，下边是一个图片数据集！ 步骤： 自定义Dataset实例： 定义 __init__ 方法：返回feature和label两个部分的数据； 定义 __ge
阅读全文
摘要：拉勾 先说好了，都是我从网上找的，我也没检查过这几张表是不是一样的，也没怎么试过。但是下边的几张表肯定存在错误，因为之前碰到过。记录一下方便查找，所以仅供参考。 1. PyTorch各版本对GPU算力要求 GPU算力：NVIDIA显卡算力 PyTorch版本 算力要求 >=1.3 >=3.5 >=1
阅读全文
