I think there is a wrong!

[rikyaa]

Training Stage with Concealing Field

if mode != 'test':
    if i_level == 0:
        accum_prod_dark = torch.cat([torch.ones_like(alpha[..., :1] * coarse_dark[0]),
            torch.cumprod(1.0 - alpha[..., :-1] + eps, dim=-1) * coarse_dark[1:]], dim=-1,)
    else:
        accum_prod_dark = torch.cat([torch.ones_like(alpha[..., :1]) * fine_dark[0],
            torch.cumprod(1.0 - alpha[..., :-1] + eps, dim=-1) * fine_dark[1:]], dim=-1,)

here :**torch.cumprod(1.0 - alpha[..., :-1] + eps, dim=-1) * fine_dark[1:]**维度是（4096，64）（128，）没办法相乘的呀？期待你的回答