Python shapes Examples

Example #1

def PrintShape(x, message='PrintShape', **unused_kwargs):
    """No-op layer that prints the shape of the stack."""
    print(message + ' ; stack shape = ' + str(base.shapes(x)))
    return x

Example #2

def PrintShape(x, message='PrintShape', **unused_kwargs):
    """No-op layer that prints the shape of the stack."""
    _print_shape(base.shapes(x), message=message)
    return x