常用的创建数据语句:
1、torch.from_numpy(ndarray) → Tensor
2、torch.ones(*sizes, out=None) → Tensor,和zeros 全0,全1,与numpy一致
3、torch.full(*sizes, fill_value: Number) → Tensor,全是fill_value
4、torch.arange(start, end, step=1, out=None) → Tensor,左闭右开,类似arange
5、torch.linspace(start, end, steps=100, out=None) → Tensor,左闭右闭,类似linspace
6、torch.normal(means, std, out=None) → Tensor,均值和方差都可以是张量和标量
torch.manual_seed(1) 设置种子,保持随机的结果一致
网友评论