Skip to content

Type: PyTorch tensor #783

@JohannesMessner

Description

@JohannesMessner

Add type TorchTensor that wraps tensors from PyTorch.

Bonus: If we can specify the desired shape like this: TorchTensor[3, 224, 224].

Metadata

Metadata

Labels

DocArray v2This issue is part of the rewrite; not to be merged into main

Type

No type

Projects

Status

Done

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions