<reponame>juanman2/mlpy
import torch

MAX_PRIME = 100
sieve = [True] * MAX_PRIME
for i in range(2, MAX_PRIME):
    if sieve[i]:
        print(i)
        for j in range(i*i, MAX_PRIME, i):
            sieve[j] = False


x = torch.rand(5, 3)
print(x)

import torch

MAX_PRIME = 100
sieve = [True] * MAX_PRIME
for i in range(2, MAX_PRIME):
    if sieve[i]:
        print(i)
        for j in range(i*i, MAX_PRIME, i):
            sieve[j] = False


x = torch.rand(5, 3)
print(x)