tensor_shape_assert
