Searched refs:tensor_getitem (Results 1 – 1 of 1) sorted by relevance
26 def tensor_getitem(inp: torch.Tensor): function31 self.checkScript(tensor_getitem, (inp,))33 scripted = torch.jit.script(tensor_getitem)