Capsule-Network-Tutorial
Capsule-Network-Tutorial copied to clipboard
Use tensor.item() for scalar conversion
To handle
IndexError:_ invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number