AHH thanks!
I realised I hadn’t reshaped my tensors when working with them so it was giving my shape errors. You’re right, once I’ve reshaped them then everything works as expected!
AHH thanks!
I realised I hadn’t reshaped my tensors when working with them so it was giving my shape errors. You’re right, once I’ve reshaped them then everything works as expected!