Skip to content

Device detection for factories.array() #2052

@ClaudiaComito

Description

@ClaudiaComito

Related
#2044 #2045 #2046

Feature functionality
ht.array(obj, device=None) should default to the object's device if obj is a torch tensor.

Currently, it defaults to CPU.

Thanks @GioPede for bringing this up.

Metadata

Metadata

Assignees

Labels

High priority, urgentbugSomething isn't working

Type

Projects

Status

In Progress

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions