reorgnize colotensor directory (#1062)

* reorgnize colotensor directory

* polish code
This commit is contained in:
Jiarui Fang
2022-06-03 18:04:22 +08:00
committed by GitHub
parent 3d10be33bd
commit a00644079e
25 changed files with 130 additions and 66 deletions

View File

@@ -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