tntorch icon indicating copy to clipboard operation
tntorch copied to clipboard

cat function does not care about device

Open SuperCrabLover opened this issue 6 months ago • 0 comments

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

SuperCrabLover avatar Dec 30 '23 17:12 SuperCrabLover