module 'torch' has no attribute 'softmax'

source_unseen_output = torch.softmax(torch.mm(source_fc8, unseen_vec.t()) * self.entropy_lambda, dim=1)
E_loss = (-source_unseen_output * source_unseen_output.log()).sum(1).mean()
loss2 = E_loss.item()
CE_loss += args.entropy_loss_weight * E_loss

File "C:/Users/Dell/Downloadstrain.py", line 158, in train
    source_unseen_output = torch.softmax(torch.mm(source_fc8, unseen_vec.t()) * self.entropy_lambda, dim=1)
AttributeError: module 'torch' has no attribute 'softmax'

检查下torch的版本

版本问题