Question

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

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]

打印张量变量

3 个答案: