pytorch
凌波漫步-
这个作者很懒,什么都没留下…
展开
-
detach()
detach()原创 2022-08-27 20:24:27 · 157 阅读 · 1 评论 -
torch.repeat()、torch.cat()、torch.max()函数
torch.repeat()、torch.cat()、torch.max()函数原创 2022-08-27 20:22:33 · 250 阅读 · 0 评论 -
torch.from_numpy()、torch.view()、torch.masked_select()、 F.softmax() 、F.log_softmax()、torch.mean()函数
torch.from_numpy()、torch.view()、torch.masked_select()、 F.softmax() 、F.log_softmax()、torch.mean()函数原创 2022-08-27 20:20:22 · 823 阅读 · 0 评论 -
numpy inf、reshape()、random.randint()、bincount()函数
numpy inf、reshape()、random.randint()、bincount()函数原创 2022-08-27 20:11:37 · 264 阅读 · 0 评论 -
Expected all tensors to be on the same device, but found at least two device, cuda: 0 and cpu!
Expected all tensors to be on the same device, but found at least two device, cuda: 0 and cpu!原创 2022-08-27 20:05:54 · 304 阅读 · 0 评论 -
pytorch 模型输出、精度计算、 构造零矩阵、修改
pytorch 模型输出、精度计算、 构造零矩阵、修改原创 2022-08-27 20:04:08 · 960 阅读 · 0 评论 -
torch 自定义dataset
torch 自定义 dataset原创 2022-08-21 16:19:21 · 213 阅读 · 0 评论 -
tensor 添加维度
tensor 添加维度原创 2022-08-21 15:00:15 · 3938 阅读 · 0 评论