pytorchtorch.cumsum和torch.cumprod的用法

68 阅读 0 评论 45 点赞

我是靠谱客的博主虚拟可乐，最近开发中收集的这篇文章主要介绍pytorchtorch.cumsum和torch.cumprod的用法，觉得挺不错的，现在分享给大家，希望可以做个参考。

torch.cumsum和torch.cumprod的用法

import torch

a = torch.arange(0,6).view(2,3)
b = torch.cumsum(a,dim=0)
c = torch.cumsum(a,dim=1)
print('a=',a)
print('b=',b)
print('c=',c)

在这里插入图片描述
dim=0时，a的第一行保持不变，从第二行开始，第二行的每个元素都是前面行的元素的累加和
dim=1时，a的第一列保持不变，从第二列开始，第二列的每个元素都是前面列的元素的累加和

import torch

a = torch.arange(0,6).view(2,3)
b = torch.cumprod(a,dim=0)
c = torch.cumprod(a,dim=1)
print('a=',a)
print('b=',b)
print('c=',c)

在这里插入图片描述
dim=0时，a的第一行保持不变，从第二行开始，第二行的每个元素都是前面行的元素的累积
dim=1时，a的第一列保持不变，从第二列开始，第二列的每个元素都是前面列的元素的累积

以上就是虚拟可乐为你收集整理的pytorchtorch.cumsum和torch.cumprod的用法的全部内容，希望文章能够帮你解决pytorchtorch.cumsum和torch.cumprod的用法所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。