python - 打印张量变量

Question

我有一个张量变量，它是一个大小为 4 * 5 的矩阵，我想知道是否有一种快速的方法可以将此变量打印到控制台。我可以使用没有 for 循环的 print 来打印这个变量吗？

Answer 1

NumPy 数组可以用很好的格式打印：

In [187]: arr = np.arange(4*5).reshape(4,5)

In [188]: print(arr)
[[ 0  1  2  3  4]
 [ 5  6  7  8  9]
 [10 11 12 13 14]
 [15 16 17 18 19]]

有关控制 NumPy 数组如何显示的更多方法，请参见numpy.set_printoptions 。

Answer 2

当然！

print ("{} {} {} {}\n" * 5).format(*t)

细节取决于张量的确切布局。在这里，我假设一个包含 20 个浮点数的元组。

或者，如果您真的很懒惰：

print (("{} "*4 + "\n")*5).format(*t)

Answer 3

def main():
    matrix = [
        [ 1,  2,  3,  4],
        [ 5,  6,  7,  8],
        [ 9, 10, 11, 12],
        [13, 14, 15, 16],
        [17, 18, 19, 20],
    ]
    print("{} {} {} {} {}".format(*matrix))

if __name__ == '__main__':
    main()

结果：

[1, 2, 3, 4] [5, 6, 7, 8] [9, 10, 11, 12] [13, 14, 15, 16] [17, 18, 19, 20]