Shortcuts

collie.utils.concat_tensor

concat_tensor(tensor_list, dim=0)[源代码]

拼接 tensor_list 中的张量,并且在拼接时将张量转移到 cpu 上来避免显存的增加。

返回:

拼接后位于 cpu 上的张量