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):
|
def layout(self):
|
||||||
return self.tensors[0].layout
|
return self.tensors[0].layout
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return f"{type(self).__name__}({self.tensors!r})"
|
||||||
|
|
||||||
|
|
||||||
def cat_nested(tensors, *args, **kwargs):
|
def cat_nested(tensors, *args, **kwargs):
|
||||||
cated_tensors = []
|
cated_tensors = []
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue