Performance improvement

[Nerdyvedi]

Following changes to improve performance slightly:

• Using .view instead of .astype to perform inplace operation which is faster

• Hardcoding output tensors indices