随笔分类 -  pytorch

记录pytorch学习的过程
摘要:一个是返回token,一个是返回其在字典中的id,如下 def bert_(): model_name = 'bert-base-chinese' MODEL_PATH = 'D:/xhzy-work/PURE/models/bert-base-chinese/' # a.通过词典导入分词器 tok 阅读全文
posted @ 2021-04-29 13:51 小千北同学超爱写代码 阅读(4020) 评论(0) 推荐(0)
摘要:pytorch里面处理的最基本的操作对象就是Tensor(张量),它就是一个多维矩阵。它和numpy唯一的不同就是,pytorch可以在GPU上运行,而numpy不可以。 Tensor的基本数据类型有五种: 32位浮点型:torch.FloatTensor。pyorch.Tensor()默认的就是这 阅读全文
posted @ 2021-04-08 10:54 小千北同学超爱写代码 阅读(391) 评论(0) 推荐(0)