Methodforward >>> Input: x_true: N, T_p, 2 x_hat: N, T_p, 2 f: N, f (128+16) >>> Output: x_hat_aft
ETH/models/layer_utils.py:263
Methodforward >>> Input: x_true: N, T_p, 2 x_hat: N, T_p, 2 f: N, f (128+16) >>> Output: x_hat_aft
models/layer_utils.py:284