标签： conv-neural-network pytorch
我编写了一个文本分割网络，当我输入单个图像时，需要1分钟的推理时间。我测试了Conv2d，它本身并不与单个批处理并行。但是，当简单地将两个张量相乘时，它将在多核中运行。 在Pytorch中上述情况是否正常，因为Tensorflow会在推理过程中并行处理单个批次