pytorch的shuffle功能

沃·夏澈德

于 2021-11-11 20:04:41 发布

阅读量3k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： pytorch学习笔记文章标签： pytorch python 数据处理

本文链接：https://blog.youkuaiyun.com/aaalswaaa1/article/details/121275076

7 篇文章

订阅专栏

这篇博客介绍了如何使用PyTorch中的torch.randperm函数生成随机索引，并展示了如何将这些索引应用到原始数据上，实现数据的随机打乱。这对于数据预处理和训练集的随机化非常重要。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import torch

# 原数据
x = torch.arange(0, 10)
print(x)

# 生成随机索引
shuffle_index=torch.randperm(10)

print(x[shuffle_index])