Shortcuts

mmocr.models.textrecog.layers.lstm_layer 源代码

# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn


[文档]class BidirectionalLSTM(nn.Module): def __init__(self, nIn, nHidden, nOut): super().__init__() self.rnn = nn.LSTM(nIn, nHidden, bidirectional=True) self.embedding = nn.Linear(nHidden * 2, nOut)
[文档] def forward(self, input): recurrent, _ = self.rnn(input) T, b, h = recurrent.size() t_rec = recurrent.view(T * b, h) output = self.embedding(t_rec) # [T * b, nOut] output = output.view(T, b, -1) return output
Read the Docs v: latest
Versions
latest
stable
0.x
dev-1.x
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.