基于numpy中的expand_dims函數(shù)用法
更新時間:2019年12月18日 15:09:47 作者:qm5132
今天小編就為大家分享一篇基于numpy中的expand_dims函數(shù)用法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
常見的一種應(yīng)用場景:
條件:假設(shè)A的shape為[4, 2],B的shape為[5, 2]
目的:實現(xiàn)A中的每一行, 減去B中的所有行(broadcast操作)。
實現(xiàn):
A1 = np.expand_dims(A, -2) => A1的shape變?yōu)閇4, 1, 2] B1 = np.expand_dims(B, 0) => B1的shape變?yōu)閇1, 5, 2] A1 - B1

其他示例:
wh = np.random.randint(1,3, size=(4,2)) np.expand_dims(wh, -2).shape np.expand_dims(wh, 1).shape
在倒數(shù)第2個軸后面(在正數(shù)第1個軸后面)插入一個新軸。



以上這篇基于numpy中的expand_dims函數(shù)用法就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
您可能感興趣的文章:
相關(guān)文章
使用Pandas實現(xiàn)高效讀取篩選csv數(shù)據(jù)
在數(shù)據(jù)分析和數(shù)據(jù)科學(xué)領(lǐng)域中,Pandas?是?Python?中最常用的庫之一,本文將介紹如何使用?Pandas?來讀取和處理?CSV?格式的數(shù)據(jù)文件,希望對大家有所幫助2024-04-04
深入理解python中sort()與sorted()的區(qū)別
Python list內(nèi)置sort()方法用來排序,也可以用python內(nèi)置的全局sorted()方法來對可迭代的序列排序生成新的序列。這篇文章主要介紹了python中sort()與sorted()的區(qū)別,需要的朋友可以參考下2018-08-08

