import torch
import time
x = torch.randn(5000, 28, 28)

t0 = time.time()
for i in range(5000):
  a = x[i, :, :]
elapsed = t0 - time.time()

library(torch)
x <- torch_randn(5000, 28, 28)
elapsed <- system.time({
  for (i in 1:5000) {
    a <- x[i,,]
  }
})

tibble::tribble(
  ~Language, ~Time,
  "R"      , lubridate::duration(elapsed[["elapsed"]]),
  "Python" , lubridate::duration(reticulate::py$elapsed)
)

## # A tibble: 2 x 2
##   Language Time              
##   <chr>    <Duration>        
## 1 R        0.888s            
## 2 Python   0.043255090713501s

x <- torch_randn(5000, 28, 28)
profvis::profvis({
  for (i in 1:5000) {
    a <- x[i,,]
  }
})