pytorch数据创建

2019-11-12  本文已影响0人  超神_484d

常用的创建数据语句:

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) 设置种子,保持随机的结果一致

上一篇下一篇

猜你喜欢

热点阅读