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'