/Python/Recurrent neural network pytorch

Recurrent neural network pytorch

import torch
from torch import nn

# check documentation in the source link for further information
rnn = nn.RNN(10, 20, 2)

# initialize vectors at t=0
x = torch.randn(5, 3, 10) 
h0 = torch.randn(2, 3, 20) 

output, hn = rnn(x, h0)

Source: pytorch.org

Similar codes

1 vote126 views1 answer
Stringbuilder python
Python
1 vote126 views2 answers
Organize order columns dataframe
Python
0 vote142 views1 answer
Graph bokeh
Python
2 votes129 views5 answers
Torch.stack
Python
0 vote116 views1 answer
Python parser txt to excel
Python
4 votes116 views3 answers
How to make python speak
Python
1 vote145 views2 answers
N-largest and n-smallest in list in python
Python
0 vote115 views1 answer
56.5 to 57 in python
Python
1 vote138 views1 answer
Create dictionary from keys and values python
Python
8 votes145 views2 answers
Python os get dir path
Python