【深度学习(李沐)】
无论使用哪个深度学习框架,它的张量类(在MXNet中为ndarray, 在PyTorch和TensorFlow中为Tensor)都与Numpy的ndarray类似。如果我们的目标形状是(高度,宽度), 那么在知道宽度后,高度会被自动计算得出,不必我们自己做除法。在上面的例子中,为了获得一个3行的矩阵,我们手动指定了它有3行和4列。下面的例子分别演示了当我们沿行(轴0,形状的第一个元素)和按列(轴1,形状的第二个元素)连结两个矩阵时,会发生什么情况。在这里,最外层的列表对应于轴0,内层的列表对应于轴1。
原创
2022-11-12 10:45:27 ·
1097 阅读 ·
1 评论