2024年7月2日
pytorch 替换softmax
摘要： import torch import torch.nn.functional as F y = F.softmax(x,dim=-1) 中的y = F.softmax(x,dim=-1)等价于y = x.exp()/torch.exp(x).sum(dim=-1, keepdim=True) 阅读全文
posted @ 2024-07-02 15:25 心比天高xzh 阅读(37) 评论(0) 推荐(0)