#pragma once // @generated by torchgen/gen.py from Function.h #include #include #include #include #include #include #include #include #include #include #include #include #include namespace at { // aten::chunk(Tensor(a -> *) self, int chunks, int dim=0) -> Tensor(a)[] inline ::std::vector chunk(const at::Tensor & self, int64_t chunks, int64_t dim=0) { return at::_ops::chunk::call(self, chunks, dim); } }