nn.BatchNorm1d, nn.Conv1d and nn.ReLU have their corresponding implements in MinkowskiEngine. However, I have trouble when I am looking for the corresponding implements of PyTorch: view( ), torch.bmm and torch.nn.Parameter(). How should I transform the code to MinkowskiEngine version?
I am trying to add an attention module into resnet, and the original torch version code is like this:
nn.BatchNorm1d, nn.Conv1d and nn.ReLU have their corresponding implements in MinkowskiEngine. However, I have trouble when I am looking for the corresponding implements of PyTorch: view( ), torch.bmm and torch.nn.Parameter(). How should I transform the code to MinkowskiEngine version?