一、
torch
下的常用API,基本上都是对数据进行处理的函数
①操作数据:
torch.is_tensor,torch.set_default_dtype,torch.get_default_dtype,torch.cat,torch.index_select,torch.reshape,torch.squeeze,torch.t,torch.unsqueeze,torch.transpose,torch.take,torch.where
②定义数据(tensor):
torch.tensor,torch.empty,torch.empty_like,torch.full,torch.full_like,torch.ones, torch.ones_like,torch.zeros,torch.zeros_like,torch.range,torch.arange,torch.line(log)space
③数据转换(主要是从ndarray转过来):
torch.from_numpy,torch.as_tensor()
④随机数:
torch.normal,torch.rand,torch.rand_like,torch.randint,torch.randint_like,torch.randn,torch.randn_like,
⑤数学计算
一般数学计算:torch.abs,torch.add,torch.clamp