Leetcode Coding Exercise On Numpy Array To Tensor