torch中的cumprod函数 含义用法示例 含义 计算tensor某个维度上元素的累计乘积 用法 形如 y = torch.cumprod(input,dim=0) 示例 import torch x = [1,2,3] #转换为tensor x = torch.tensor(x) y = torch.cumprod(x,dim=0) print(y) #output: tensor([1, 2, 6])