pytorch學習(一)

1. 張量的創建和操作   創建爲初始化矩陣,並初始化 a = torch.empty(5, 3) #創建一個5*3的未初始化矩陣 nn.init.zeros_(a) #初始化a爲0 nn.init.constant_(a, 3) # 初始化a爲3 nn.init.uniform_(a) #初始化爲uniform分佈   隨機數矩陣 torch.rand(5, 3)
相關文章
相關標籤/搜索