Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Better way to deal with "lists" of tensors #116

Open
reverendbedford opened this issue Dec 14, 2023 · 0 comments
Open

Better way to deal with "lists" of tensors #116

reverendbedford opened this issue Dec 14, 2023 · 0 comments
Assignees
Labels
C++ enhancement New feature or request help wanted Extra attention is needed

Comments

@reverendbedford
Copy link
Collaborator

reverendbedford commented Dec 14, 2023

We need to consider a third set of dimensions in our classes that are set at runtime. As an example, a BatchTensor might have shape (3,4) + (7,) + (6,) where (3,4) is the batch dimension, (7,) is the "list" dimension, and (6,) is the base dimension. So we could view this as a list of batched SR4 tensors, for example. By default all our current mathematical operators (except the list tensor stuff) will treat the "list" dimension as batch dimension. But we can add new outer product type operators that apply binary operators to do list outer productions. So if A is shape (3,4) + (7,) + (6,) and B is shape (3,4) + (5,) + (6,) list_product(A, B, [](auto a, auto b){return a+b;}) would return a tensor with shape (3,4) + (7,5) + (6,) that adds the list entries of A and B together in all outer product combinations.

@hugary1995 hugary1995 added enhancement New feature or request help wanted Extra attention is needed labels Jun 24, 2024
@hugary1995 hugary1995 self-assigned this Jun 24, 2024
@hugary1995 hugary1995 added the C++ label Jun 24, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
C++ enhancement New feature or request help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

2 participants