Open GUESSZERO4 opened 2 years ago
输入的Size(2,32,2000)在模型输出后的Size为什么还是(2,9)。它的第一个维度2代表着什么?还是batch吗。输出不应该是对于几个类别的预测概率吗?
输入size的含义为[batch, channel, seq_len],输出size的含义为[batch, num_classes],至于第一个维度2就是代表的batch size。
输入的Size(2,32,2000)在模型输出后的Size为什么还是(2,9)。它的第一个维度2代表着什么?还是batch吗。输出不应该是对于几个类别的预测概率吗?