Pytorch
文章平均质量分 55
记录学习pytorch的一些问题
Rssevenn
这个作者很懒,什么都没留下…
展开
-
【Pytorch】torch.squeeze和torch.unsqueeze函数
【Pytorch】torch.squeeze和torch.unsqueezetorch.squeeze(input, int)返回一个张量,其可以在原来的input的shape基础之上,压缩维度,比如输入的input的shape为torch.Size([31, 1, 8]),调用torch.squeeze(input, 1),意思就是将原来的input的第二列**(0,1,2三列)**“删除”,返回的tensor的shape为 torch.Size([31, 8])注意点 int的范围:假设原来sh原创 2022-03-29 11:35:07 · 1811 阅读 · 0 评论 -
【Pytorch】torch.reshape函数
【Pytorch】torch.reshapetorch.reshape(input, shape)返回一个张量,其与input的元素和元素个数相同。>>> a = torch.arange(4.)>>> torch.reshape(a, (2, 2))tensor([[ 0., 1.], [ 2., 3.]])>>> b = torch.tensor([[0, 1], [2, 3]])>>> torch原创 2022-03-29 11:04:16 · 4720 阅读 · 0 评论 -
【Pytorch】基于GRU和LSTM的时间序列数据预测实现
利用pytorch框架下的gru和lstm进行时间序列的预测文末附源代码文件原创 2022-03-26 23:38:15 · 8080 阅读 · 20 评论 -
【Pytorch】从0开始的数据集制作,从numpy数组到Dataset,再到Dataloader
从0创建一个Datalader类的数据集完整代码行行注释原创 2022-03-25 22:33:13 · 5482 阅读 · 0 评论 -
OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.已解决
pytorch下的libiomp5md.dll冲突问题原创 2021-11-30 21:46:58 · 1153 阅读 · 0 评论