Manage your account and access personalized content. Sign up for an Oracle Account
Access your cloud dashboard, manage orders, and more. Free Cloud Platform Trial
The batch dimension is at the wrong place in the mentioned tensor.
Ensure that the batch dimension is the first dimension in the tensor.