Tesla Interview Question

Implement softmax classification for CNN