tf.contrib가 tensorflow 2.0으로 오면서 지원이 중단되었고, 해당 클래스에 있는 메소드는 어떻게 대체되는지 설명해주실 수 있을까요?

다른 클래스는 어떻게 대체되었는지 tensorflow 측에서 git에 공지를 해주었는데 contrib는 제가 못찾는건지, 그냥 지원이 중단되었다고만 적혀있네요..

여기서는 tfe.Iterator를 어떻게 대체해야할지 모르겠어요.

EPOCHS = 1001

for step in range(EPOCHS):
    for features, labels  in tfe.Iterator(dataset):
        grads = grad(logistic_regression(features), features, labels)
        optimizer.apply_gradients(grads_and_vars=zip(grads,[W,b]))
        if step % 100 == 0:
            print("Iter: {}, Loss: {:.4f}".format(step, loss_fn(logistic_regression(features),features,labels)))
test_acc = accuracy_fn(logistic_regression(x_test),y_test)
print("Testset Accuracy: {:.4f}".format(test_acc))

도와주시면 감사하겠습니다.