Solved by anantguptadbl in post #2 
              
              
                @theGuyWithBlackTie This looks more of a numpy problem rather than a pytorch problem 
e.g. 
import time
X = None
batch_size=1024
embedding_size=5000
for i in range(100):
    start_time = time.time()
    if X is None:
        X = np.random.rand(batch_size, embedding_size)
    else:
        X = np.app…