Searched defs:tensor_split (Results 1 – 3 of 3) sorted by relevance
5097 def tensor_split(input, indices_or_sections, axis=0): function
3659 def tensor_split(input, indices_or_sections, axis=0): function
3749 def tensor_split(self, indices_or_sections, axis=0): member in Tensor