Skip to content

Instantly share code, notes, and snippets.

@mbednarski
Created March 6, 2018 22:40
Show Gist options
  • Save mbednarski/53fae1e526258734980737cea0fa6ae7 to your computer and use it in GitHub Desktop.
Save mbednarski/53fae1e526258734980737cea0fa6ae7 to your computer and use it in GitHub Desktop.
embedding_dims = 5
W1 = Variable(torch.randn(embedding_dims, vocabulary_size).float(), requires_grad=True)
z1 = torch.matmul(W1, x)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment