A.- A._, née en 1988, domiciliée à D._, est atteinte d'une surdité sévère bilatérale d'origine congénitale. Pour lui permettre de suivre l'enseignement de l'école publique, un traitement de logopédie, pris en charge par l'assurance-invalidité fédérale, lui est dispensé à E._ par une praticienne spécialisée, chez laquelle elle se rend en moyenne une fois tous les quinze jours. Par décision du 28 mai 1999, l'Office de l'assurance-invalidité pour le canton de Vaud (ci-après : l'OAI) a refusé de continuer à prendre en charge les frais de transport liés à ce traitement, dès le 1er mai 1999. B.- Par jugement du 30 octobre 2000, le Tribunal des assurances du canton de Vaud a rejeté le recours formé contre cette décision par A._. C.- Cette dernière, représentée par son père, interjette recours de droit administratif contre ce jugement en concluant, avec suite de frais et dépens, à son annulation, à l'octroi de subsides pour les frais de transport liés au traitement logopédique et au renvoi de la cause à l'administration afin qu'elle en détermine le montant. L'OAI a renoncé à se déterminer cependant que l'Office fédéral des assurances sociales (OFAS) conclut au rejet du recours.