【PyTorch】 初始化張量方式
-
生成全0矩陣
# 矩陣大小為2x4,數據類型為32bit整數型 torch.zeros( [2,4], dtype=torch.int32 )
-
生成全1矩陣
torch.ones( [2,4] )
-
生成對角矩陣
# 矩陣大小為3x3 torch.eye(3)
-
生成未初始化的指定大小矩陣
# 矩陣大小為2x3 torch.empty(2,3)
-
利用指定值填充矩陣
# 用 6 填充矩陣 torch.full([2,3], fill_value=6)
-
生成等距向量
# 值的範圍 [start, end),不含 end # step: 步伐大小,每次增加 0.2 torch.arange( start=0, end=1, step=0.2)
# 值的範圍 [start, end],包含 end # steps: 步伐數量,[0,1]之間切成等距的五步 torch.linspace( start=0, end=1, steps=5)
-
生成 [0, 1] 均勻隨機採樣矩陣
# 矩陣大小為 2x5 torch.rand(2,5)
-
生成標準正態分布採樣矩陣
torch.randn(3,4)
-
生成範圍 [low, high) 隨機採樣整數矩陣
torch.randint(low=0, high=20, size=(3,4))