Closed tzuhanchang closed 1 week ago
Before the fix, tensors can only be concatenated over dim=0. The dim argument was not used by any operation in the function. This update allows the tensors to be concatenated over any given dimension.
dim=0
dim
Before the fix, tensors can only be concatenated over
dim=0
. Thedim
argument was not used by any operation in the function. This update allows the tensors to be concatenated over any given dimension.