- 新建Tensor的常用方法
函数 | 功能 |
Tensor(*sizes) | 基础构造函数 |
ones(*sizes) | 全1的Tensor |
zeros(*sizes) | 全0的Tensor |
eye(*sizes) | 对角线为1,其他为0的Tensor |
arange(s,e,step) | 从s到e,步长为step的一维Tensor |
linspace(s,e,steps) | 从s到e,均匀切分为steps份 |
rand/randn(*sizes) | 均匀/标准分布 |
normal(mean,std)/uniform(from,to) | 正太分布/标准分布 |
randperm(m) | 随机排列 |