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)表示二维数据,可以进行矩阵乘法
如果加到矩阵上的话,默认是加到每一列