1. Memory usage, consider replacing reshape functions to view 2. Consider local operation instead of copying tensors