def METHOD_NAME(module, input, output, name):
    # y_mini_batch_outputs.append(output)
    # y_mini_batch_inputs.append(input[0])
    y_mini_batch_outputs[name] = output
    y_mini_batch_inputs[name] = input[0]