2024.6.21 powの計算速度の比較【numpy, torch】

numpy, torchが提供するpow関数の計算速度を比較する。

code:pow1.py

import numpy as np

import torch as pt

import time

loopmax = 100000000

x = 3.14

#

x_np = np.float64(x)

time1_init = time.process_time()

for i in range(loopmax):

y = np.pow(x_np, 2)

time1 = time.process_time() - time1_init

print('numpy.pow :', time1)

#

x_pt = pt.tensor(x, dtype=pt.float)

time2_init = time.process_time()

for i in range(loopmax):

y = pt.pow(x_pt, 2)

time2 = time.process_time() - time2_init

print('torch.pow :', time2)

試行 numpy torch

１回目 55.1864 130.6523

２回目 51.7767 123.6804

３回目 52.2358 134.3302

torch.powはnumpy.powの倍以上の計算時間を要している。