将指定Tensor的数据存入文件“ofile”,若“ofile”为None,则直接将tensor数据进行屏显。
def print_tensor(tensor, ofile=None):
参数名 |
类型 |
说明 |
---|---|---|
tensor |
Tensor |
需要被打印的Tensor。 |
ofile |
文件对象或流。 可选参数 |
指定的写入tensor数据的文件,若不配置,默认为当前系统的stdout(输出到屏幕)。 |
无
无
请参见调试示例。