# logits [batch_size, frame_length, vocab_size]
logits = np.load('logits_array.npy')
logits = torch.from_numpy(logits.transpose((1,0))).requires_grad_()
报错:ValueError: axes don’t match array
logits = torch.from_numpy(logits.transpose((1,0,2))).requires_grad_()
解决。