2021年3月14日
pytorch nll_loss batch shape
摘要： 使用nll_loss时，如果想计算batch的loss，假设loss函数输入x的shape为 (N, d, C)，其中N是batch_size，d是句子长度，C是vocab_size，标签target y的shape为(N, d)。 nll_loss函数要求输入为 (N, C, d)，target为 阅读全文
posted @ 2021-03-14 15:45 AliceYing 阅读(160) 评论(0) 推荐(0)