torch.cumprod¶
- torch.cumprod(input, dim, *, dtype=None, out=None) Tensor ¶
返回
input
在维度dim
上元素的累积乘积。例如,如果
input
是大小为 N 的向量,则结果也将是大小为 N 的向量,其元素为。- 参数
- 关键字参数
dtype (
torch.dtype
, 可选) – 返回张量的期望数据类型。如果指定,则在执行操作之前将输入张量强制转换为dtype
。这对于防止数据类型溢出很有用。默认值:None。out (Tensor, 可选) – 输出张量。
示例
>>> a = torch.randn(10) >>> a tensor([ 0.6001, 0.2069, -0.1919, 0.9792, 0.6727, 1.0062, 0.4126, -0.2129, -0.4206, 0.1968]) >>> torch.cumprod(a, dim=0) tensor([ 0.6001, 0.1241, -0.0238, -0.0233, -0.0157, -0.0158, -0.0065, 0.0014, -0.0006, -0.0001]) >>> a[5] = 0.0 >>> torch.cumprod(a, dim=0) tensor([ 0.6001, 0.1241, -0.0238, -0.0233, -0.0157, -0.0000, -0.0000, 0.0000, -0.0000, -0.0000])