Numpy中take的用法
@numpy.take今天在学习批量梯度下降到时候,遇到到了numpy.take的使用。在踩了不少坑之后把正确的用法分享给大家,希望有所帮助。首先:import numpy as np#一维a.take([1,3,5])然后检索并返回索引值a=np.array([1,2,3,4,5,6,7,8])a.take([1,3,5])Out: array([2, 4, 6])a.take([[1,2],[5,6],[7,3]])Out[: array([[2, 3], [6, 7
原创
2020-06-17 15:48:35 ·
2681 阅读 ·
4 评论