关于batchsize,重中之重 这个教程很关键

https://blog.csdn.net/alwaysyxl/article/details/83050137

import numpy as np

# a = np.array([[[1, 2]], [[3, 4]]])
a = np.array([[1, 2], [3, 4]])
print(a.shape)
new_a = a.reshape(2, 1, 2)
print(new_a)
print("转换后的维度是",new_a.shape)
print("===" * 20)

b = np.array([[[1, 2, 3], [1, 1, 1]], [[4, 5, 6], [7, 8, 9]]])
print("a的维度", a.shape)
print("b的维度", b.shape)
c = np.matmul(new_a, b)
print("c的维度", c.shape)
print(c)