转载自: https://blog.csdn.net/qq_22210253/article/details/85229988
调试pytorch的交叉熵损失函数,看到输入input和目标target的维度shape不一样,很诧异,就在网上看到了这篇文章,就明白了。
转载自: https://blog.csdn.net/qq_22210253/article/details/85229988
调试pytorch的交叉熵损失函数,看到输入input和目标target的维度shape不一样,很诧异,就在网上看到了这篇文章,就明白了。