PyTorch NLP 常用 API
根据官方文档整理,知道有这么些常用的函数,具体要用的时候再去官方文档里面查询数据创建由列表、numpy创建或tensor自定义的随机整数、正态分布等import torchimport numpy as np# *****tensor创建*****device = torch.device('cuda:0')a = np.array([1, 2, 3])t1 = torch.from_numpy(a) # 原地创建tensort2 = torch.as_tensor(a) # .
原创
2020-06-08 22:59:28 ·
309 阅读 ·
0 评论