maybe just have ndim>=3 raise for now (or do an apply)