numpy 和 pytorch 对数组操作的比较

a = np.array([1, 2])

b = tf.tensor([1, 2])

c = np.array([[1, 2]])

d = tf.tensor([[1, 2]])

e = np.array([[1], [2]])

f = tf.tensor([[1], [2]])

模块：numpy 数据类型：adarray，查看数据的维度：a.ndim ——dim表示dimension

模块：pytorch 数据类型：tensor， 查看数据的维度：b.dim

查看数据的维度主要看有几个中括号

a和b都是一维数据，a.shape() ——返回（2，）表示一维数据，不能进行矩阵乘法

如果加到矩阵上的话，默认是加到每一行

c和d都是二维数据，a.shape() ——返回（1，2）表示二维数据，可以进行矩阵乘法

如果加到矩阵上的话，默认是加到每一行

e和f都是二维数据，a.shape() ——返回（2，1）表示二维数据，可以进行矩阵乘法

如果加到矩阵上的话，默认是加到每一列