从CrossEntropyLoss的源码的描述中可以看到答案:
红框中的minibatch表示batch_size, C表示class_num,后面可以跟着其他维度。
这样就可以使用3维或者更高维的变量!!!
京公网安备 11010502049817号