随机数在深度学习中扮演着重要的角色,而PyTorch提供了多种随机数生成函数来满足不同的需求。本文将详细介绍PyTorch中的四个常用随机数生成函数:randn、normal、rand和uniform_,并提供相应的源代码示例。
-
randn函数:
randn函数用于生成服从标准正态分布(均值为0,标准差为1)的随机数。它的使用方法如下:import torch # 生成一个形状为(3, 4)的张量,元素服从标准正态分布 x = torch.randn(3, 4) print(x
随机数在深度学习中扮演着重要的角色,而PyTorch提供了多种随机数生成函数来满足不同的需求。本文将详细介绍PyTorch中的四个常用随机数生成函数:randn、normal、rand和uniform_,并提供相应的源代码示例。
randn函数:
randn函数用于生成服从标准正态分布(均值为0,标准差为1)的随机数。它的使用方法如下:
import torch
# 生成一个形状为(3, 4)的张量,元素服从标准正态分布
x = torch.randn(3, 4)
print(x