OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
173f126068
pytorch/torch/nn
History
zeshengzong c9c0f8eae3 Add plot for torch.nn.Threshold and torch.nn.GLU (#150171) 
Fixes #150170

## Changes

- Add plot for `torch.nn.Threshold` and `torch.nn.GLU`
- Add example output make them easier get result by users

## Test Result

![image](https://github.com/user-attachments/assets/f6c5bc46-f9b7-4db7-9797-e08d8423d1b3)

![image](https://github.com/user-attachments/assets/ad4e6c84-7b29-44f1-b7bd-9c81e4a92ef8)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150171
Approved by: https://github.com/albanD
2025-04-08 03:55:37 +00:00
..
attention [FlexAttention] Remove dead code (#150575) 2025-04-03 01:46:19 +00:00
backends
intrinsic
modules Add plot for torch.nn.Threshold and torch.nn.GLU (#150171) 2025-04-08 03:55:37 +00:00
parallel [BE][Ez]: Use itertools.chain.from_iterable when possible (#148190) 2025-03-06 20:37:06 +00:00
qat
quantizable
quantized
utils Add hint message for pack_padded_sequence (#146747) 2025-02-20 06:27:07 +00:00
__init__.py
_reduction.py
common_types.py [5/N] Apply Ruff fixes and pyupgrade to Python 3.9 (#144205) 2025-01-15 04:00:47 +00:00
cpp.py
functional.py Fix torch.nn.functional.one_hot param num_classes optional description (#146470) 2025-02-06 07:48:05 +00:00
functional.pyi.in Use Python 3.9 typing (#148157) 2025-03-04 03:09:55 +00:00
grad.py
init.py
parameter.py remove typo in UninitializedParameter docstring (#140197) 2024-11-15 23:26:23 +00:00
parameter.pyi remove allow-untyped-defs from torch/nn/parameter.pyi (#144654) 2025-01-13 19:02:31 +00:00