【pytorch】在多个batch中如何使用nn.CrossEntropyLoss

NoSuchKey