pytorch conv2d计算shape,prefetch_factor提速，提高显卡利用率

N = (W − F + 2P )/S+1

• N  ：  output_shape 为 N x N

• W ：  input_shape 为 W×W

• F  ：  Filter 大小 F×F

• P  ：  Padding 大小

• S  ：  步长 stride

kernel_size

  卷积核的大小，一般我们会使用5x5、3x3这种左右两个数相同的卷积核，因此这种情况只需要写kernel_size = 5这样的就行了。如果左右两个数不同，比如3x5的卷积核，那么写作kernel_size = (3, 5)，注意需要写一个tuple，而不能写一个列表（list）

DataLoader(dataset, batch_size=1, shuffle=False, sampler=None,
           batch_sampler=None, num_workers=0, collate_fn=None,
           pin_memory=False, drop_last=False, timeout=0,
           worker_init_fn=None, *, prefetch_factor=2,
           persistent_workers=False)