cerebras.modelzoo.common.run_cstorch_flow.log_input_summary#

cerebras.modelzoo.common.run_cstorch_flow.log_input_summary(batch)[source]#

Log the input tensors to tensorboard.