numpy不同维度数组相加,或者叫张量相加?

import numpy as np

a=np.array([[1,2],[3,4]])
b=np.array([[[7,8],[9,10]],
           [[5,6],[7,8]]])
c=b-a
print(c)
print(b.shape)
print(c.shape)
d=b+a
print(d)
print(d.shape)

运行得到的结果是:

[[[6 6]
  [6 6]]

 [[4 4]
  [4 4]]]
(2, 2, 2)
(2, 2, 2)
[[[ 8 10]
  [12 14]]

 [[ 6  8]
  [10 12]]]
(2, 2, 2)

可见,张量相加,类似于把低维张量扩充成高维张量,再加减。乘除估计类似。谢谢大家 

猜你喜欢

转载自blog.csdn.net/anlongstar/article/details/130546792
今日推荐