- 博客(3)
- 收藏
- 关注
RSS订阅原创 Python中torch的gather函数使用详解
假设我们向神经网络一次性输入多个样本，则输出返回也是多个样本，记录在一个张量中。其中输出的维度为4，表示每次输入会输出4个元素，batch_size=3，表示样本数量为3，我们一次性输入了3个样本。使用gather函数有两个参数，一个是dim（维度），另一个是index（索引表）。dim=0表示批量维度，即数据中的样本数量，dim=1表示输出层维度。此时必须与相对应，因为此时，第一行的[0]表示对应输出层第一行的第[0]个输出元素，第二行的[1]表示对应输出层第二行的第[1]个输出元素...
2025-12-23 16:05:27
136
原创 python中调用collections创建队列和基本的随机采样操作
k表示一次采样得到5个元素，输出sample得到：[[9, 4], [7, 7], [8, 1], [9, 0], [8, 5]]到达maxlen后，队列会自动采取舍弃先进的元素，如此时[0,0],[0,1],...等元素被弹出。
2025-12-19 17:25:45
103
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人