Lines Matching full:arange

733             "list_literal_of_tensor": "[torch.arange(3), torch.arange(5)]",
735             "list_literal_of_mixed": "[torch.arange(5), 1]",
736 …   "list_comprehension_of_tensor": "[torch.add(x, 1) for x in [torch.arange(3), torch.arange(5)]]",
738             "list_comprehension_of_mixed": "[torch.add(1, x) for x in [torch.arange(5), 1]]",
872             "dict_literal_of_str_tensor": '{"foo" : torch.arange(3), "bar" : torch.arange(5)}',
874             "dict_literal_of_mixed": '{"foo" : torch.arange(3), "bar" : 2}',
876                     zip(["foo", "bar"], [torch.arange(3), torch.arange(5)])}',
880                     zip(["foo", "bar"], [torch.arange(3), 2])}',
881             "dict_keyword": "dict(foo=torch.arange(3), baz=torch.arange(5))",
882 …        "dict_keyword_with_iterable": 'dict([("foo", torch.arange(3)), ("bar", torch.arange(5))])',
884 …_with_internal_aggregate_function": 'dict(zip(["foo", "bar"], [torch.arange(3), torch.arange(5)])',
885             "dict_keyword_with_mapping": 'dict({"foo" : torch.arange(3), "bar" : torch.arange(5)})',
886 …rd_with_mapping_and_kwargs": 'dict({"foo" : torch.arange(3), "bar" : torch.arange(5)}, baz=torch.a…