Make it easier to debug nested tensors. (#15383)
This commit is contained in:
parent
43cb4fffc8
commit
cd84f47efe
|
|
@ -83,6 +83,9 @@ class NestedTensor:
|
|||
def layout(self):
|
||||
return self.tensors[0].layout
|
||||
|
||||
def __repr__(self):
|
||||
return f"{type(self).__name__}({self.tensors!r})"
|
||||
|
||||
|
||||
def cat_nested(tensors, *args, **kwargs):
|
||||
cated_tensors = []
|
||||
|
|
|
|||
Loading…
Reference in New Issue