ssjdoudou的博客

Pytorch中的nn.LSTMPytorch中LSTM总共有7个参数，前面3个是必须输入的 input_size– The number of expected features in the inputx hidden_size– The number of features in the hidden stateh num_layers– Number...

有个很好的解决办法在程序最开始的地方加上device = torch.device("cuda" if torch.cuda.is_available() else "cpu")使用的时候self.encoder = EncoderAtt(input_size=self.X.shape[1], hidden_size=encoder_hidden_size, T=T).to(d...

输入数据pytorch的input:(seq_len, batch, input_size)三维的Tensor，大概长这样(450, 2, 6)450是训练集的长度，2是时间步长，这个数据你要自己去转化，6是特征数对应股票数据就是（训练集的长度，一次是几条数据，股票的特征）原始的数据集是二维的，(450，6)，你需要展开成三维的(450,1,6)，作为pytorch的输入，...

今天看到一行pytorch的代码import torchfrom torch.autograd import Variabletensor = torch.FloatTensor([[1,2],[3,4]]) variable = Variable(tensor, requires_grad=True) v_out = torch.mean(variable*va...

本文将用Pytorch构建一个最简单的线性神经网络，练习Pytorch中神经网络模型的保存和重载定义一个线性神经网络y=wx+by = wx + by=wx+b这是一个基本的网络Net，它只包含一个全连接层import torchfrom torch import nnclass Net(nn.Module): def __init__(self): s...

本文主要介绍pytoch和numpy的数据之间的转化和pytorch中张量tensor的使用使用numpy新建一个一维数组import torchimport numpy as npnp_data = np.arange(6)print(np_data,type(np_data))[0 1 2 3 4 5] <class 'numpy.ndarray'>把数组转化为...