标签：郭哥 tensor 代码 torch list 余弦 num time np

import numpy as np
import torch
import time

a_list = []

for i in range(32):
        a = np.ones((27,512))
        a_list.append(a)

t0 = time.time()
mode = torch.zeros((len(a_list),27,512))
for i in range(100):
        for j,num in enumerate(a_list):
                num_tensor = torch.from_numpy(num)
                mode[j] = num_tensor
        # num_np = np.stack(a_list, axis=0)
        # num_tensor = torch.from_numpy(num_np)
        # num_tensor = torch.Tensor(a_list)
t1 = time.time()

print(mode.shape,num_tensor.shape, (t1-t0)/100)

import numpy as np
import torch
import time

a_list = []

for i in range(32):
        a = np.zeros((27,512))
        a_list.append(a)

t0 = time.time()

for i in range(100):
        j = 0
        for num in a_list:
                num_tensor = torch.from_numpy(num)
                if j == 0:
                        first_tensor = torch.reshape(num_tensor, (1, 27, 512))
                elif j == 1:
                        second_tensor = torch.reshape(num_tensor, (1, 27, 512))
                        new_tensor = torch.cat((second_tensor, first_tensor), 0)
                else:
                        num_tensor = torch.reshape(num_tensor, (1, 27, 512))
                        new_tensor = torch.cat((new_tensor, num_tensor), 0)
                j += 1

        # num_np = np.stack(a_list, axis=0)
        # num_tensor = torch.from_numpy(num_np)
        # num_tensor = torch.Tensor(a_list)
t1 = time.time()
print(new_tensor.shape, t1-t0)

标签：郭哥,tensor,代码,torch,list,余弦,num,time,np
来源： https://blog.csdn.net/weixin_43489941/article/details/120880100

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。