Find better strategies to generate values in tensors

The current strategy uses np.arange which causes some issues with larger numbers. Plus it only generates integer values. Number generate strategy should be configurable.