w = torch.empty(3, 5) >>> nn.init.uniform_(w) torch.nn.init.normal_(tensor, mean=0, std=1). 用正态分布 初始化输入 Tensor 。 参数:. tensor – n 维 torch.
確定! 回上一頁