higgsfield / Capsule-Network-Tutorial

Pytorch easy-to-follow Capsule Network tutorial
767 stars 135 forks source link

Passing index as variable in index_select() #3

Open himalayajung opened 6 years ago

himalayajung commented 6 years ago

Passing index as a variable instead of a tensor in index_select()