def shape_from_args(self): """Returns the shape of the index expression. """ return u.Shape(*ku.size(self.key, self.args[0]._dcp_attr.shape))
def size_from_args(self): """Returns the shape of the index expression. """ return ku.size(self.key, self.args[0].size)