摘要: 一、torch.nn.Embedding 模块可以看做一个字典,字典中每个索引对应一个词和词的embedding形式。利用这个模块,可以给词做embedding的初始化操作 torch.nn.Embedding(num_embeddings, embedding_dim, padding_idx=N 阅读全文
posted @ 2021-01-31 18:31 光彩照人 阅读(503) 评论(0) 推荐(0) 编辑
摘要: torch.utils.data.Dataset与torch.utils.data.DataLoader的理解 pytorch提供了一个数据读取的方法,其由两个类构成:torch.utils.data.Dataset和DataLoader 我们要自定义自己数据读取的方法,就需要继承torch.uti 阅读全文
posted @ 2021-01-31 10:40 光彩照人 阅读(410) 评论(0) 推荐(0) 编辑