Assigning slice of a tensor to a variable? - Modular