栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

np。将ND张量/阵列与一维数组连接

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

np。将ND张量/阵列与一维数组连接

您可以使用

np.repeat

r = np.concatenate((a, b.reshape(1, 1, -1).repeat(a.shape[0], axis=0)), axis=1)

它的作用是,首先调整

b
数组的形状以匹配的尺寸
a
,然后根据
a
的第一个轴根据需要重复其值多次:

b3D = b.reshape(1, 1, -1).repeat(a.shape[0], axis=0)array([[[1, 2, 3]],       [[1, 2, 3]],       [[1, 2, 3]],       [[1, 2, 3]],       [[1, 2, 3]]])b3D.shape(5, 1, 3)

然后将此中间结果与

a
-

r = np.concatenate((a, b3d), axis=0)r.shape(5, 5, 3)

这与您当前的答案有所不同,主要是因为值的重复不是硬编码的(即,重复处理了它)。

如果您需要针对其他数量的尺寸(而不是3D阵列)处理此问题,则需要进行一些更改(主要是删除的硬编码重塑形状

b
)。


时机

a = np.random.randn(100, 99, 100)b = np.random.randn(100)# Tai's answer%timeit np.insert(a, 4, b, axis=1)100 loops, best of 3: 3.7 ms per loop# Divakar's answer%%timeit b3D = np.broadcast_to(b,(a.shape[0],1,len(b)))np.concatenate((a,b3D),axis=1)100 loops, best of 3: 3.67 ms per loop# solution in this post%timeit np.concatenate((a, b.reshape(1, 1, -1).repeat(a.shape[0], axis=0)), axis=1)100 loops, best of 3: 3.62 ms per loop

这些都是极具竞争力的解决方案。但是,请注意性能取决于实际数据,因此请确保先进行测试!



转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/669454.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号