pytorch API
仲夏199603
这个作者很懒,什么都没留下…
展开
-
pytorch的pack和unpack函数
packunpack导入from torch.nn.utils.rnn import pack_padded_sequence as packfrom torch.nn.utils.rnn import pad_packed_sequence as unpack使用方法举例 embs = pack(self.embedding(inputs), lengths) outputs, ...原创 2018-10-19 10:03:43 · 3157 阅读 · 1 评论 -
torch的GLU函数
import torch.nn as nnnn.GLU()GLU门控线性单元的描述API原创 2018-10-19 12:26:19 · 7324 阅读 · 0 评论