.
dimension: {steak.shape}") steak = tf.expand_dims(steak, axis=0) # add an extra dimension at axis 0 #steak = steak[tf.newaxis, ...] # alternative to the above, '...' is short for 'every other dimension' print(f"Shape after new dimension: {steak.shape}") steak Our custom image has a batch size of 1!