Avatar notebook default
2篇文章 · 1002字 · 1人关注
  • PyTorch:数据读取机制DataLoader

    先明确几个常见的名词含义:batch、epoch、iterationbatch:通常我们将一个数据集分割成若干个小的样本集,然后一小部分一小部分...

  • PyTorch:切片函数index_select()

    index_select()函数有两种用法。第一种是将被切片的函数作为参数传入index_select()中 还有一种是调用张量内置的index...

文集作者