當前位置: 首頁>>代碼示例>>Python>>正文


Python Models.EncoderBase方法代碼示例

本文整理匯總了Python中onmt.Models.EncoderBase方法的典型用法代碼示例。如果您正苦於以下問題:Python Models.EncoderBase方法的具體用法?Python Models.EncoderBase怎麽用?Python Models.EncoderBase使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在onmt.Models的用法示例。


在下文中一共展示了Models.EncoderBase方法的1個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。

示例1: forward

# 需要導入模塊: from onmt import Models [as 別名]
# 或者: from onmt.Models import EncoderBase [as 別名]
def forward(self, input, lengths=None, hidden=None):
        """ See :obj:`onmt.modules.EncoderBase.forward()`"""
        self._check_args(input, lengths, hidden)

        emb = self.embeddings(input)
        s_len, batch, emb_dim = emb.size()

        emb = emb.transpose(0, 1).contiguous()
        emb_reshape = emb.view(emb.size(0) * emb.size(1), -1)
        emb_remap = self.linear(emb_reshape)
        emb_remap = emb_remap.view(emb.size(0), emb.size(1), -1)
        emb_remap = shape_transform(emb_remap)
        out = self.cnn(emb_remap)

        return emb_remap.squeeze(3).transpose(0, 1).contiguous(),\
            out.squeeze(3).transpose(0, 1).contiguous() 
開發者ID:xiadingZ,項目名稱:video-caption-openNMT.pytorch,代碼行數:18,代碼來源:Conv2Conv.py


注:本文中的onmt.Models.EncoderBase方法示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台,相關代碼片段篩選自各路編程大神貢獻的開源項目,源碼版權歸原作者所有,傳播和使用請參考對應項目的License;未經允許,請勿轉載。