How to use the einops.einops.parse_shape function in einops
To help you get started, we’ve selected a few einops examples, based on popular ways it is used in public projects.
Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
arogozhnikov / einops / tests / test_other.py View on Github 
print('Shape parsing for ', backend.framework_name)
x = numpy.zeros([10, 20, 30, 40])
parsed1 = parse_shape(x, 'a b c d')
parsed2 = parse_shape(backend.from_numpy(x), 'a b c d')
assert parsed1 == parsed2 == dict(a=10, b=20, c=30, d=40)
assert parsed1 != dict(a=1, b=20, c=30, d=40) != parsed2
parsed1 = parse_shape(x, '_ _ _ _')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ _')
assert parsed1 == parsed2 == dict()
parsed1 = parse_shape(x, '_ _ _ hello')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ hello')
assert parsed1 == parsed2 == dict(hello=40)
parsed1 = parse_shape(x, '_ _ a1 a1a111a')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ a1 a1a111a')
assert parsed1 == parsed2 == dict(a1=30, a1a111a=40)def test_parse_shape_imperative():
backends = collect_test_backends(symbolic=False, layers=False)
backends += collect_test_backends(symbolic=False, layers=True)
for backend in backends:
print('Shape parsing for ', backend.framework_name)
x = numpy.zeros([10, 20, 30, 40])
parsed1 = parse_shape(x, 'a b c d')
parsed2 = parse_shape(backend.from_numpy(x), 'a b c d')
assert parsed1 == parsed2 == dict(a=10, b=20, c=30, d=40)
assert parsed1 != dict(a=1, b=20, c=30, d=40) != parsed2
parsed1 = parse_shape(x, '_ _ _ _')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ _')
assert parsed1 == parsed2 == dict()
parsed1 = parse_shape(x, '_ _ _ hello')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ hello')
assert parsed1 == parsed2 == dict(hello=40)
parsed1 = parse_shape(x, '_ _ a1 a1a111a')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ a1 a1a111a')
assert parsed1 == parsed2 == dict(a1=30, a1a111a=40)def test_parse_shape_imperative():
backends = collect_test_backends(symbolic=False, layers=False)
backends += collect_test_backends(symbolic=False, layers=True)
for backend in backends:
print('Shape parsing for ', backend.framework_name)
x = numpy.zeros([10, 20, 30, 40])
parsed1 = parse_shape(x, 'a b c d')
parsed2 = parse_shape(backend.from_numpy(x), 'a b c d')
assert parsed1 == parsed2 == dict(a=10, b=20, c=30, d=40)
assert parsed1 != dict(a=1, b=20, c=30, d=40) != parsed2
parsed1 = parse_shape(x, '_ _ _ _')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ _')
assert parsed1 == parsed2 == dict()
parsed1 = parse_shape(x, '_ _ _ hello')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ hello')
assert parsed1 == parsed2 == dict(hello=40)
parsed1 = parse_shape(x, '_ _ a1 a1a111a')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ a1 a1a111a')
assert parsed1 == parsed2 == dict(a1=30, a1a111a=40)def test_parse_shape_imperative():
backends = collect_test_backends(symbolic=False, layers=False)
backends += collect_test_backends(symbolic=False, layers=True)
for backend in backends:
print('Shape parsing for ', backend.framework_name)
x = numpy.zeros([10, 20, 30, 40])
parsed1 = parse_shape(x, 'a b c d')
parsed2 = parse_shape(backend.from_numpy(x), 'a b c d')
assert parsed1 == parsed2 == dict(a=10, b=20, c=30, d=40)
assert parsed1 != dict(a=1, b=20, c=30, d=40) != parsed2
parsed1 = parse_shape(x, '_ _ _ _')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ _')
assert parsed1 == parsed2 == dict()
parsed1 = parse_shape(x, '_ _ _ hello')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ hello')
assert parsed1 == parsed2 == dict(hello=40)
parsed1 = parse_shape(x, '_ _ a1 a1a111a')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ a1 a1a111a')
assert parsed1 == parsed2 == dict(a1=30, a1a111a=40)def test_parse_shape_imperative():
backends = collect_test_backends(symbolic=False, layers=False)
backends += collect_test_backends(symbolic=False, layers=True)
for backend in backends:
print('Shape parsing for ', backend.framework_name)
x = numpy.zeros([10, 20, 30, 40])
parsed1 = parse_shape(x, 'a b c d')
parsed2 = parse_shape(backend.from_numpy(x), 'a b c d')
assert parsed1 == parsed2 == dict(a=10, b=20, c=30, d=40)
assert parsed1 != dict(a=1, b=20, c=30, d=40) != parsed2
parsed1 = parse_shape(x, '_ _ _ _')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ _')
assert parsed1 == parsed2 == dict()
parsed1 = parse_shape(x, '_ _ _ hello')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ hello')
assert parsed1 == parsed2 == dict(hello=40)
parsed1 = parse_shape(x, '_ _ a1 a1a111a')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ a1 a1a111a')
assert parsed1 == parsed2 == dict(a1=30, a1a111a=40)def test6(x):
# parsing parameters
t = rearrange(x, 'b c h w -> (b h w) c')
t = t[:, ::2] # replacement for dot-product, just changes size of second axis
assert t.shape == (10 * 30 * 40, 10)
y = rearrange(t, '(b h w) c2 -> b c2 h w', **parse_shape(x, 'b _ h w'))
assert y.shape == (10, 10, 30, 40)
return yfor backend in backends:
if backend.framework_name == 'keras':
# need special way to compile, shape vars can be used only inside layers
continue
print('special shape parsing for', backend.framework_name)
input_symbols = [
backend.create_symbol([10, 20, 30, 40]),
backend.create_symbol([10, 20, None, None]),
backend.create_symbol([None, None, None, None]),
]
if backend.framework_name in ['mxnet.symbol']:
# mxnet can't normally run inference
input_symbols = [backend.create_symbol([10, 20, 30, 40])]
for input_symbol in input_symbols:
shape_placeholder = parse_shape(input_symbol, 'a b c d')
shape = {}
for name, symbol in shape_placeholder.items():
shape[name] = symbol if isinstance(symbol, int) \
else backend.eval_symbol(symbol, [(input_symbol, numpy.zeros([10, 20, 30, 40]))])
print(shape)
result_placeholder = rearrange(input_symbol, 'a b (c1 c2) (d1 d2) -> (a b d1) c1 (c2 d2)',
**parse_shape(input_symbol, 'a b c1 _'), d2=2)
result = backend.eval_symbol(result_placeholder, [(input_symbol, numpy.zeros([10, 20, 30, 40]))])
print(result.shape)
assert result.shape == (10 * 20 * 20, 30, 1 * 2)
assert numpy.allclose(result, 0)def test_parse_shape_imperative():
backends = collect_test_backends(symbolic=False, layers=False)
backends += collect_test_backends(symbolic=False, layers=True)
for backend in backends:
print('Shape parsing for ', backend.framework_name)
x = numpy.zeros([10, 20, 30, 40])
parsed1 = parse_shape(x, 'a b c d')
parsed2 = parse_shape(backend.from_numpy(x), 'a b c d')
assert parsed1 == parsed2 == dict(a=10, b=20, c=30, d=40)
assert parsed1 != dict(a=1, b=20, c=30, d=40) != parsed2
parsed1 = parse_shape(x, '_ _ _ _')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ _')
assert parsed1 == parsed2 == dict()
parsed1 = parse_shape(x, '_ _ _ hello')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ hello')
assert parsed1 == parsed2 == dict(hello=40)
parsed1 = parse_shape(x, '_ _ a1 a1a111a')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ a1 a1a111a')
assert parsed1 == parsed2 == dict(a1=30, a1a111a=40)x = numpy.zeros([10, 20, 30, 40])
parsed1 = parse_shape(x, 'a b c d')
parsed2 = parse_shape(backend.from_numpy(x), 'a b c d')
assert parsed1 == parsed2 == dict(a=10, b=20, c=30, d=40)
assert parsed1 != dict(a=1, b=20, c=30, d=40) != parsed2
parsed1 = parse_shape(x, '_ _ _ _')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ _')
assert parsed1 == parsed2 == dict()
parsed1 = parse_shape(x, '_ _ _ hello')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ hello')
assert parsed1 == parsed2 == dict(hello=40)
parsed1 = parse_shape(x, '_ _ a1 a1a111a')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ a1 a1a111a')
assert parsed1 == parsed2 == dict(a1=30, a1a111a=40)backends = collect_test_backends(symbolic=False, layers=False)
backends += collect_test_backends(symbolic=False, layers=True)
for backend in backends:
print('Shape parsing for ', backend.framework_name)
x = numpy.zeros([10, 20, 30, 40])
parsed1 = parse_shape(x, 'a b c d')
parsed2 = parse_shape(backend.from_numpy(x), 'a b c d')
assert parsed1 == parsed2 == dict(a=10, b=20, c=30, d=40)
assert parsed1 != dict(a=1, b=20, c=30, d=40) != parsed2
parsed1 = parse_shape(x, '_ _ _ _')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ _')
assert parsed1 == parsed2 == dict()
parsed1 = parse_shape(x, '_ _ _ hello')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ _ hello')
assert parsed1 == parsed2 == dict(hello=40)
parsed1 = parse_shape(x, '_ _ a1 a1a111a')
parsed2 = parse_shape(backend.from_numpy(x), '_ _ a1 a1a111a')
assert parsed1 == parsed2 == dict(a1=30, a1a111a=40)einops
A new flavour of deep learning operations
Package Health Score
87 / 100Popular einops functions
- einops._backends.AbstractBackend
- einops._backends.AbstractBackend.__subclasses__
- einops._backends.get_backend
- einops._backends.UnknownSize
- einops.asnumpy
- einops.einops._prepare_transformation_recipe
- einops.einops._reductions
- einops.einops.EinopsError
- einops.einops.parse_shape
- einops.einops.rearrange
- einops.einops.reduce
- einops.einops.TransformRecipe
- einops.layers.keras.keras_custom_objects
- einops.layers.keras.Rearrange
- einops.layers.keras.Reduce
- einops.layers.RearrangeMixin
- einops.layers.ReduceMixin
- einops.layers.torch.Rearrange
- einops.layers.torch.Reduce
- einops.rearrange