mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-21 09:29:47 +00:00
reorgnize colotensor directory (#1062)
* reorgnize colotensor directory * polish code
This commit is contained in:
@@ -1,10 +1,8 @@
|
||||
import torch
|
||||
|
||||
from colossalai.tensor.colo_tensor import ColoTensor
|
||||
|
||||
from typing import Iterator, Tuple, Union
|
||||
import torch.nn as nn
|
||||
from colossalai.tensor import ColoTensor
|
||||
from colossalai.tensor.colo_tensor import ColoTensor
|
||||
|
||||
|
||||
# The function is credited to PyTorch Team
|
||||
|
Reference in New Issue
Block a user