Fix seed shape issue

[scharalambous-sony]

• Check for seed shape fails when number of seeds and batch size are 1
• Modify unit test to capture the bug