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)
可见,张量相加,类似于把低维张量扩充成高维张量,再加减。乘除估计类似。谢谢大家