NishchalPrasad
Results
1
comments of
NishchalPrasad
You need to squeeze the tensor in dimensions 1 and 2. It should work then.