娇气小海豚

文章
2
资源
0
加入时间
3年0月9天

Pytorch mask:上三角和下三角

上三角 triuPytorch上三角和下三角的调用与numpy是相同的。np.triu(np.ones((5,5)),k=0) # k控制对角线开始的位置Out[25]: array([[1., 1., 1., 1., 1.], [0., 1., 1., 1., 1.], [0., 0., 1., 1., 1.], [0., 0., 0., 1., 1.], [0., 0., 0., 0., 1.]])构建一个上三角masktorch.t