Pytorch Dataloader内部随机数

pytorch的dataloader的__getitem__()方法,内部定义随机数,采用numpy.random.rand的方法,每次迭代从dataloader里取数据,都会产生拿到相同的随机数序列,相反,torch.rand的方法,每次迭代拿数据时,拿到的是不同的随机数序列。

	# numpy生成随机数
    def get_train_sample(self, ):
        track_idx = np.random.randint(0, len(self._tracks), (1, ))[0]
        separated = self._tracks[track_idx]
        patch_length = int(self._sample_length * self._sample_rate)
        if self._incoherent_rate <= np.random.rand(1)[0]:   # coherent remix
            start_t = np.random.randint(0, separated.shape[-1]-patch_length, (1, ))
            separated = separated[:, :, start_t: start_t+patch_length]
		return separated
	# torch生成随机数
    def get_train_sample(self, ):
        track_idx = torch.randint(0, len(self._tracks), (1,)).item()
        separated = self._tracks[track_idx]
        patch_length = int(self._sample_length * self._sample_rate)
        if self._incoherent_rate <= np.random.rand(1)[0]:   # coherent remix
            start_t = torch.randint(0, separated.shape[-1]-patch_length, (1, )).item()
            separated = separated[:, :, start_t: start_t+patch_length]
		return separated

用第一种方法生成的数据片段,每个loader里面都相同,用第二种方法,每个loader依旧是随机。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
PyTorch 是一个基于 Python 的科学计算库,它主要针对两类人群:1、NumPy 的替代品;2、深度学习研究平台,提供了丰富的工具和接口,支持快速实验开发。 以下是 PyTorch 中常用函数的手册: 1. 张量操作函数 - torch.Tensor:创建一个张量 - torch.zeros:创建一个全是零的张量 - torch.ones:创建一个全是一的张量 - torch.rand:创建一个随机的张量 - torch.eye:创建一个单位矩阵 - torch.randn:从标准正态分布中抽取随机数 - torch.from_numpy:将 NumPy 数组转化为张量 - torch.arange:创建一个从 start 到 end,步长为 step 的序列张量 - torch.linspace:创建一个从 start 到 end,共有 steps 个数的序列张量 2. 数学函数 - torch.add:张量加法 - torch.sub:张量减法 - torch.mul:张量乘法 - torch.div:张量除法 - torch.pow:求幂 - torch.sqrt:求平方根 - torch.exp:求指数 - torch.log:求对数 - torch.abs:求绝对值 - torch.max:求最大值 - torch.min:求最小值 - torch.sum:求和 - torch.mean:求平均值 - torch.std:求标准差 3. 神经网络函数 - torch.nn.Linear:线性变换 - torch.nn.Conv2d:二维卷积 - torch.nn.MaxPool2d:二维最大池化 - torch.nn.ReLU:ReLU 激活函数 - torch.nn.Sigmoid:Sigmoid 激活函数 - torch.nn.Tanh:Tanh 激活函数 - torch.nn.Dropout:随机失活正则化 4. 数据处理函数 - torch.utils.data.Dataset:自定义数据集 - torch.utils.data.DataLoader:数据加载器 - torchvision.transforms:数据预处理 - torchvision.datasets:常用数据集 这些函数只是 PyTorch 中常用的一部分,PyTorch 还提供了很多其他的函数和工具,可以根据不同的需求选择使用。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值