diff --git a/spanner/batch.go b/spanner/batch.go
index e1ed21f23cbc..87696c2e2bbf 100644
--- a/spanner/batch.go
+++ b/spanner/batch.go
@@ -449,3 +449,11 @@ func (p *Partition) UnmarshalBinary(data []byte) error {
 	}
 	return err
 }
+
+// GetPartitionToken returns partition token
+func (p *Partition) GetPartitionToken() []byte {
+	if p != nil {
+		return p.pt
+	}
+	return nil
+}