weixin_41522231的博客

原创 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 查看数据的维度主要看有几个中括