Searched defs:dsplit (Results 1 – 4 of 4) sorted by relevance
1030 dsplit = _split_on_axis('dsplit', axis=2) variable
278 def dsplit(ary: ArrayLike, indices_or_sections): function
2614 std::vector<Tensor> dsplit(const Tensor& self, int64_t split_size) { in dsplit() function2664 std::vector<Tensor> dsplit(const Tensor& self, IntArrayRef split_sizes) { in dsplit() function
4445 def dsplit(a: TensorLikeType, sections: DimsType) -> TensorSequenceType: function