![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
代码调试
Dumbledore‘s wand
这个作者很懒,什么都没留下…
展开
-
【代码记录】pytorch矩阵取数据--避免for循环
import torchinput = [ [2, 3, 4, 5, 0, 0], [1, 4, 3, 0, 0, 0], [4, 2, 2, 5, 7, 0], [1, 0, 0, 0, 0, 0]]torch.gather(torch.tensor(input),1,torch.tensor([[3],[2],[4],[0]]))注意点 input 和index 需要转成tensorParametersinput (Tensor) – the sourc原创 2022-02-23 11:09:09 · 2201 阅读 · 0 评论 -
【代码记录】从数据集中取10条数据
pytorch从数据集中取10条数据进行调试dataClass.validIdList = dataClass.validIdList[:10]dataClass.validSampleNum = len(dataClass.validIdList)dataClass.trainIdList = dataClass.validIdListdataClass.trainSampleNum = len(dataClass.trainIdList)...原创 2022-02-21 23:08:01 · 787 阅读 · 0 评论