forked from ccf-ai-infra/GPUCodeForces
fixes pairwisedistance #38
This commit is contained in:
parent
79b57f276f
commit
eda2f35259
|
|
@ -5,6 +5,7 @@ You have complete freedom to choose the set of operators you want to replace. Yo
|
|||
Here's an example to show you the syntax of inline embedding custom CUDA operators in torch: The example given architecture is:
|
||||
|
||||
```python
|
||||
pairwisedistance.py
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
import torch.nn.functional as F
|
||||
|
|
|
|||
Loading…
Reference in New Issue