Print操作命令
该OP创建一个打印操作,打印正在访问的Tensor内容。
封装传入的Tensor,以便无论何时访问Tensor,都会打印信息message和Tensor的当前值。
参数:
input (Variable)-将要打印的Tensor
first_n (int)-打印Tensor的次数
print_tensor_name (bool)-可选,指明是否打印Tensor名称,默认为True
print_tensor_shape (bool)-可选,指明是否打印Tensor维度信息,默认为True
返回:输出Tensor
注解
输入和输出是两个不同的Variable,在接下来的过程中,应该使用输出Variable而非输入Variable,否则打印层将失去backward的信息。
代码示例: