Add horizon dimension to problem variables

[itsikelis]

Make all variables torch tensors (instead of a list) to allow vectorization capabilities.