A = torch.tensor([2,4,5,6])
B = torch.cumprod(A,-1)
C = torch.cumsum(A,-1)
print(B)
print(C)
输出:
tensor([ 2, 8, 40, 240])
tensor([ 2, 6, 11, 17])
Process finished with exit code 0
A = torch.tensor([2,4,5,6])
B = torch.cumprod(A,-1)
C = torch.cumsum(A,-1)
print(B)
print(C)
输出:
tensor([ 2, 8, 40, 240])
tensor([ 2, 6, 11, 17])
Process finished with exit code 0