【PyTorch】 初始化張量方式

【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))