kaniblu/torch_example7.py

Created August 20, 2020 11:10

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/kaniblu/9876d5a266d1fb66cb72d794e78bae93.js"></script>
Save kaniblu/9876d5a266d1fb66cb72d794e78bae93 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

torch_example7.py

torch.einsum("blm,mn->bln", torch.randn(3, 4, 5), torch.randn(5, 6)).size()  # torch.Size([3, 4, 6])

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment