list to torch

Posted 2023-02-18 雪靡

list to torch

方法一：

直接转换

torch.tensor(pick, dtype=torch.long)

方法二：

numpy转换

pick = numpy.array(pick)
torch.from_numpy(pick).type(torch.LongTensor)

两种方法测试下来，在list长度过大的情况下，间接转换效率高；但在list长度少的情况下，使用直接转换效率高。

测试代码：

import random
import sys
import time
from datetime import datetime

import numpy
import torch

# 有gpu的情况下使用gpu
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')



total_time = 0
"""
测试模式，0表示直接转换，1表示使用numpy转换
"""
test_mode = 1
loop_time = 100
list_size = 64000
if test_mode == 0:

    for _ in range(loop_time):
        pick = [random.randint(0, 10000) for _ in range(list_size)]
        begin_time = time.perf_counter()
        torch.tensor(pick, dtype=torch.long).to(device)
        end_time = time.perf_counter()
        total_time += end_time - begin_time
else:
    for _ in range(loop_time):
        pick = [random.randint(0, 10000) for _ in range(list_size)]
        pick = numpy.array(pick)
        begin_time = time.perf_counter()
        torch.from_numpy(pick).type(torch.LongTensor).to(device)
        end_time = time.perf_counter()
        total_time += end_time - begin_time
print("%.5fms" % (total_time * 1000.0))