program error

[s-mason]

'# calc T_c: [batch_size, 10]
# T_c = Y, is my understanding correct? Try it.
T_c = self.Y
# [batch_size, 10], element-wise multiply
L_c = T_c * max_l + cfg.lambda_val * (1 - T_c) * max_r '

Sorry, you understanded wrong .You can read https://medium.com/@pechyonkin/part-iv-capsnet-architecture-6a64422f7dce .
The second expression should be a matrix, and contains nine elements.