网络编程
位置:首页>> 网络编程>> Python编程>> 基于numpy中的expand_dims函数用法

基于numpy中的expand_dims函数用法

作者:qm5132  发布时间:2023-02-25 23:24:20 

标签:numpy,expand,dims

常见的一种应用场景:

条件:假设A的shape为[4, 2],B的shape为[5, 2]

目的:实现A中的每一行, 减去B中的所有行(broadcast操作)。

实现:


A1 = np.expand_dims(A, -2) => A1的shape变为[4, 1, 2]

B1 = np.expand_dims(B, 0) => B1的shape变为[1, 5, 2]

A1 - B1

基于numpy中的expand_dims函数用法

其他示例:


wh = np.random.randint(1,3, size=(4,2))

np.expand_dims(wh, -2).shape

np.expand_dims(wh, 1).shape

在倒数第2个轴后面(在正数第1个轴后面)插入一个新轴。

基于numpy中的expand_dims函数用法

基于numpy中的expand_dims函数用法

基于numpy中的expand_dims函数用法

来源:https://blog.csdn.net/qm5132/article/details/83379171

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com