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