10 lines
162 B
Python
10 lines
162 B
Python
import torch.nn as nn
|
|
|
|
class Identity(nn.Module):
|
|
def __init__(self, ):
|
|
super(Identity, self).__init__()
|
|
|
|
def forward(self, x):
|
|
return x
|
|
|