Equivariance with Learned Canonicalization Functions