code: func shapeEq(a, b []int) bool {
  	if isScalar(a) && isScalar(b) {
  		return true
  	}
  
  	if len(a) != len(b) {
  		return false
  	}
  
  	for i, v := range a {
  		if b[i] != v {
  			return false
  		}
  	}
  	return true
  }
docs: // shapeEq is adapted from tensor