Skip to content

cat function does not care about device #46

@SuperCrabLover

Description

@SuperCrabLover

If iterable "ts" consists of tntorch.tensors with core on cuda, the algo "cat" will fall with error due to creating and using "torch.zeros" on cpu by default. https://tntorch.readthedocs.io/en/latest/_modules/tools.html#cat

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions