|
|
|
|
|
by nuisance-bear
1893 days ago
|
|
Here's an example that illustrates the phenomenon. If memory serves me right, index latency is superlinear in dimension count. import time, torch
from itertools import product
N = 100
ten = torch.randn(N,N,N)
arr = ten.numpy()
def indexTimer(val):
start = time.time()
for i,j,k in product(range(N), range(N), range(N)):
x = val[i, j, k]
end = time.time()
print('{:.2f}'.format(end-start))
indexTimer(ten)
indexTimer(arr)
|
|