How to use 'torch permute' in Python
Every line of 'torch permute' code snippets is scanned for vulnerabilities by our powerful machine learning engine that combs millions of open source libraries, ensuring your Python code is secure.
All examples are scanned by Snyk Code
By copying the Snyk Code Snippets you agree to
49 def _permute(input, perm=None): 50 ctx = MakeContext(inputs=[input]); n_perm = len(perm) if perm else 0 51 key = 'torch.ops.permute/{}:{}/n_perm:{}'.format(ctx[0], ctx[1], n_perm) 52 module = get_module(Permute, key, ctx, n_perm=n_perm) 53 return module.forward(input, perm)
52 @add_module_test(torch.float32, torch.device('cuda'), [(1, 3, 4, 5, 6)]) 53 def test_permute_list(): 54 return Permute([0, 4, 1, 3, 2])