nestedtensor icon indicating copy to clipboard operation
nestedtensor copied to clipboard

Published 20 hours ago • verified publisher icon pytorch
→

Metadata

[Prototype] Tools for the concurrent manipulation of variably sized Tensors.

Results 49 nestedtensor issues
Sort by recently updated
recently updated
newest added

Add more tests for squeeze

more irregular nested_sizes

izdeby avatar izdeby

testing

Revisit .size() to support narrow along nested dimension

izdeby avatar izdeby

enhancement

Fix NestedTensor.nested_size() to print torch.Size() for constituents and construct torch.Sizes() on generation.

[See](https://github.com/pytorch/nestedtensor/pull/35/files#diff-550bf0adbe042646464e9b7ac2c2773fR72)

izdeby avatar izdeby

enhancement

Move NestedTensor.iter to C++

izdeby avatar izdeby

enhancement

Move NestedTensor.to to C++

izdeby avatar izdeby

enhancement

Uncomment is_contiguous() and element_size() checks in assertEqual

izdeby avatar izdeby

bug
testing

enable assertNotEqual in testing

izdeby avatar izdeby

enhancement
testing

Enable codecoverage

One way to do this could be by porting our tests and test gen to pytest.

cpuhrsch avatar cpuhrsch

enhancement
testing

Masking follow up

- exclude scalar case from get_max_size - update test cases for test_ntftm_test_multi_tensor_mix_mask3 (remove empty tensor) - use nested_size() instead of unbind() in get_max_size - look into getting rid of tolist()

izdeby avatar izdeby

enhancement

Add pickle support

See more info [here](https://github.com/pytorch/nestedtensor/issues/14).

izdeby avatar izdeby

enhancement

← Metadata

251
Stars
28
Forks
Watchers

Owner

verified publisher icon pytorch

Metadata

[Prototype] Tools for the concurrent manipulation of variably sized Tensors.

Back