matplotlib tensorflow