Pytorch
衣带渐宽人憔悴
学习笔记
展开
-
torch.nn.LSTM pytorch BiLSTM Mnist
import torchimport torch.nn as nnimport torchvision.transforms as transformsimport torchvision.datasets as dsetsfrom torch.autograd import Variable'''STEP 1: LOADING DATASET'''train_dataset = dsets.MNIST(root='./data', .原创 2021-06-16 10:39:30 · 1424 阅读 · 3 评论 -
OSError: libcudart.so.11.0: cannot open 完美解决
之前一直以为是这个问题:https://blog.csdn.net/qq_35294564/article/details/115461626原创 2021-06-16 09:36:43 · 17589 阅读 · 1 评论 -
Pytorch tensor的复制函数torch.repeat_interleave()和torch.repeat()
repeat_interleave(self: Tensor, repeats: _int, dim: Optional[_int]=None)参数说明:self: 传入的数据为tensorrepeats: 复制的份数dim: 要复制的维度,可设定为0/1/2.....>>> a=torch.Tensor([[1,2,3,4],[2,3,4,5]])>>> >>> >>> a.repeat_interleav原创 2021-04-25 14:48:05 · 3627 阅读 · 0 评论