mirror of
https://github.com/wassname/pytorch-for-numpy-users.git
synced 2026-06-27 16:10:21 +08:00
@@ -118,7 +118,7 @@
|
||||
| `x.argmin` | `mins, indices = torch.min(x, [dim])` |
|
||||
| `x.max` | `maxs, indices = torch.max(x, [dim])` |
|
||||
| `x.argmax` | `maxs, indices = torch.max(x, [dim])` |
|
||||
| `x.clip` | |
|
||||
| `x.clip` | `x.clamp` |
|
||||
| `x.round` | `x.round` |
|
||||
| `np.floor(x)` | `x.floor()` |
|
||||
| `np.ceil(x)` | `x.ceil()` |
|
||||
|
||||
+1
-1
@@ -135,7 +135,7 @@ constructors:
|
||||
- numpy: x.argmax
|
||||
pytorch: maxs, indices = torch.max(x, [dim])
|
||||
- numpy: x.clip
|
||||
pytorch:
|
||||
pytorch: x.clamp
|
||||
- numpy: x.round
|
||||
pytorch: x.round
|
||||
- numpy: np.floor(x)
|
||||
|
||||
Reference in New Issue
Block a user