时间序列实战(预测sin函数)
RNN层结构
导入包
import numpy as np
import torch
import torch.nn as nn
import torch.optim as optim
from matplotlib import pyplot as plt
构建RnnNet
- torch.nn.Module.parameters() 用法
torch.nn.RNN(input, h_0) # outputs: output, h_n
torch.nn.RNN() 用法
class Net(nn.Module):
def __init__(self, ):
super(Net, self).__init__()
self.rnn = nn.RNN(