![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
转角遇高数
自然语言处理初学者
展开
-
## pytorch中view和permute函数改变张量维度方式的不同
pytorch中view和permute函数改变张量维度方式的不同说的有点太抽象了。结合代码自己理解一下吧。。实验代码:import torchimport numpy as npa=np.array([ [ [1,2,3], [4,5,6] ]...原创 2019-09-04 20:19:06 · 722 阅读 · 0 评论 -
PyTorch中损失函数NLLLoss与CrossEntropyLoss区别
实验代码:import torchfrom torch import nninput=torch.randn(3,3)print(input)sm=nn.Softmax(dim=1)loss1=torch.nn.NLLLoss()target=torch.tensor([0,2,1])print(loss1(torch.log(sm(input)),target))los...原创 2019-09-11 19:20:30 · 2090 阅读 · 0 评论