Skip to content

Track more functions as TENSOR_CONSTRUCTORS#91

Merged
wdhongtw merged 1 commit into
mainfrom
fix-construct
Apr 28, 2026
Merged

Track more functions as TENSOR_CONSTRUCTORS#91
wdhongtw merged 1 commit into
mainfrom
fix-construct

Conversation

@wdhongtw
Copy link
Copy Markdown
Collaborator

Add linspace and logspace.

import torch
import torchax

torchax.enable_globally()
tensor = torch.linspace(3, 10, 5, device=torch.device('cpu'))
print(tensor.device)

Before the fix

jax:0

After

cpu

Add linspace and logspace.

Signed-off-by: Weida Hong <wdhongtw@google.com>
@wdhongtw wdhongtw merged commit 9ce5f2d into main Apr 28, 2026
5 checks passed
@wdhongtw wdhongtw deleted the fix-construct branch April 28, 2026 01:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants