Searched defs:tensor_split (Results 1 – 3 of 3) sorted by relevance
139 def tensor_split( function
995 std::vector<Tensor> tensor_split(const Tensor& self, IntArrayRef indices, int64_t dim) { in tensor_split() function1003 std::vector<Tensor> tensor_split(const Tensor& self, const Tensor& tensor_indices_or_sections, int6… in tensor_split() function
4106 def tensor_split( function