Show Menu
Cheatography

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)