Searched defs:unsafe_split (Results 1 – 4 of 4) sorted by relevance
124 def unsafe_split( function
1052 def unsafe_split( function
1444 def unsafe_split(input: Tensor, split_size: int, dim: int = 0) -> Tuple[Tensor, ...]: function
2588 std::vector<Tensor> unsafe_split(const Tensor& self, int64_t split_size, int64_t dim) { in unsafe_split() function