【debug】ValueError: axes don‘t match array 解决

# 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_()

解决。