nestedtensor icon indicating copy to clipboard operation
nestedtensor copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

nested_tensor constructor should be able to accept mixed input

Open izdeby opened this issue 5 years ago • 0 comments 

a = nt.nested_tensor([
            nt.nested_tensor([]),
            nt.nested_tensor([
                torch.tensor([1])
            ])
        ])

same for cuda
should succeed

a = nt.nested_tensor([
           [],
           [
               torch.tensor([1])
           ]
       ])

izdeby avatar Apr 14 '20 18:04 izdeby