/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

0 vote198 views1 answer
Python - Cómo comprobar si dos cuerdas son anagramas
Python
2 votes202 views1 answer
Jinja inheritance
Python
3 votes217 views2 answers
Python append value to dictionary list
Python
-2 votes232 views2 answers
Python find lcm
Python
0 vote232 views1 answer
Python get unique pairs from two lists
Python
0 vote255 views1 answer
Newrelic api for Tags
Python
0 vote239 views2 answers
Closures in python
Python
0 vote245 views1 answer
Bar plot fix lenthgy labels matplot
Python
0 vote227 views1 answer
Verbose python
Python
0 vote205 views1 answer
How to use selenium on default chrome python
Python