Lines Matching full:tensor_split

400         a_split_dim0 = a.tensor_split(7, 0)
403         a_split_dim1 = a.tensor_split(7, 1)
1839                         result1 = torch.tensor_split(a, sections, dim)
1840                         result2 = torch.tensor_split(
1883                         result_1 = torch.tensor_split(a, indices, dim)
1884                         result_2 = torch.tensor_split(
1910                 r"tensor_split expected at least a 1-dimensional tensor, "
1920                 r"tensor_split expected at least a 1-dimensional tensor, "
1945                 torch.tensor_split(a, sections_or_indices, dim)
1947                 torch.tensor_split(a, torch.tensor(sections_or_indices), dim)
1951         # addtional tests for tensor_split with tensor_indices_or_sections
1954 …          r"tensor_split expected tensor_indices_or_sections to have dtype of long, but got Float",
1956             torch.tensor_split(a, torch.tensor(1.1), dim)
1960             r"tensor_split expected tensor_indices_or_sections to be a"
1963             torch.tensor_split(torch.rand(S, device=device), torch.tensor(((1,),)), 0)