Show Menu
Cheatography
  1. Home >
  2. Cheat Sheets >

Neural Networks Cheat Sheet (DRAFT) by

Cheatsheet for Neural Networks

This is a draft cheat sheet. It is a work in progress and is not finished yet.

BASIC NETWORK SPECS

Embedd­ing­(VO­CAB­_SIZE, EMBEDD­ING­_DIM)

BASIC NETWORK SPECS

Embedd­ing­(VO­CAB­_SIZE, EMBEDD­ING­_DIM)

Embedding

Input - 
[]
 
nn.Emb­edd­ing­(VO­CAB­_SIZE, EMBEDD­ING­_DIM)

Embedding

Input - [SEQ_LEN, BATCH_­SIZE]
Output - 
[SEQ_LEN, BATCH_­SIZE, EMBEDD­ING_DIM
 
nn.Emb­edd­ing­(VO­CAB­_SIZE, EMBEDD­ING­_DIM)