![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
Miss.wei
这个作者很懒,什么都没留下…
展开
-
torch.split()
所以,torch.split()是一个很有用的函数,可以轻松地将张量分割成任意形状和大小的张量列表,以用于后续处理。官网解释:Splits the tensor into chunks.——PyTorch中用于分割张量的函数。官网链接:https://pytorch.org/docs/stable/torch.html。作用:将一个多维张量分割成多个张量。原创 2023-06-06 20:10:20 · 5299 阅读 · 4 评论 -
torch.cumsum()
总之,torch.cumsum()是一个简单但非常实用的函数,可以计算张量的元素累积和,应用在许多领域中。行1加到行2: 1 + 4 = 5;2 + 5 = 7;列2加到列3: 3 + 3 = 6;9 + 6 = 15。列1加到列2: 1 + 2 = 3;作用:用于计算张量中的元素累积和。行1不变:1,2,3。原创 2023-06-06 18:48:23 · 878 阅读 · 0 评论 -
torch.triu
总之,torch.triu()是一个用于获取矩阵上三角或下三角部分的简单但非常有用的函数,可以用于遮挡、矢量化和矩阵操作等目的。作用:用于获取矩阵的上三角部分。取主对角线之上的元素。取对角线之上的元素。取对角线之下的元素。原创 2023-06-06 16:52:02 · 1329 阅读 · 0 评论