如何使用print完整打印一个field的全部内容

综合讨论区
#1

使用print进行debug时需要查看大矩阵（用field存储）的完整内容，但是打印结果是缩略的：
image

请问如何使用print完整打印一个field的全部内容

#2

Hi,

我的想法是可以写个ti.kernel函数手动print出来

import taichi as ti

ti.init(ti.cpu)
A = ti.Vector.field(2, dtype=ti.f32, shape=(16, 16))

@ti.kernel
def set_field(in_A: ti.template()):
    for I in ti.grouped(in_A):
        in_A[I] = I

@ti.kernel
def print_field(in_A: ti.template()):
    for I in ti.grouped(in_A):
        print(I, ": ", in_A[I])

if __name__ == "__main__":
    set_field(A)
    print_field(A)
1 个赞