tntorch
tntorch copied to clipboard
cat function does not care about device
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