您当前的位置: 首页 >  Python

Better Bench

暂无认证

  • 1浏览

    0关注

    695博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

【Python-Numpy】numpy.stack()的解析与使用

Better Bench 发布时间:2021-01-27 10:44:19 ,浏览量:1

作用

在给定的轴下,增加或改变张量数组的维度 一般有四个轴 x = 0,x = 1,x = 2, x = -1 假设要转变的张量数组arrays的长度为N,其中的每个张量数组的形状为(A, B) 如果轴axis=0,则转变后的张量的形状为(N, A, B) 如果轴axis=1,则转变后的张量的形状为(A, N, B) 如果轴axis=2,则转变后的张量的形状为(A, B, N) 如果轴axis=-1,则转变后的张量的形状为(B, A)

举例使用

(1)如果轴axis=0,则转变后的张量的形状为(N, A, B)

arrays = [np.random.randn(3, 4) for _ in range(10)]
np.stack(arrays, axis=0).shape

(10, 3, 4)

(2)如果轴axis=1,则转变后的张量的形状为(A, N, B,)

np.stack(arrays, axis=1).shape

(3, 10, 4)

(3)如果轴axis=2,则转变后的张量的形状为(A, B, N)

np.stack(arrays, axis=2).shape

(3, 4, 10)

(4)如果轴axis=-1,张量实现转置

a = np.array([1, 2, 3])
b = np.array([2, 3, 4])
np.stack((a, b)).shape
(2,3)
np.stack((a, b), axis=-1).shape
(3,2)
关注
打赏
1665674626
查看更多评论
立即登录/注册

微信扫码登录

0.0473s