#
# returns Y of same shape as X, with
#
- # Y[:,t] = A[:,0] * Y_init + X[:,0] if t == 0
- # = A[:,t] * Y[:,t-1] + X[:,t] otherwise
+ # Y[:, t] = A[:, 0] * Y_init + X[:, 0] if t == 0
+ # = A[:, t] * Y[:, t-1] + X[:, t] otherwise
@staticmethod
def forward(ctx, A, X, Y_init):