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)
相关文章
相关标签/搜索