zeshengzong fdc8ccc5bc Make Adam, AdamW work with nonzero-dim Tensor betas (#149939) ...

Fixes #147921

## Changes

- Convert tensor `betas` using `_to_scalar`
- Change annotation of `betas` param
- Change param type in docs

## Test Result

```bash
pytest -s test/test_optim.py -k test_tensor_lr -vv
```

![image](https://github.com/user-attachments/assets/312ee045-1e8b-4789-aa6e-ba63e6df7e81)

![image](https://github.com/user-attachments/assets/7e6ec274-645b-46b9-b1a6-2b340a685203)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/149939
Approved by: https://github.com/janeyx99

Co-authored-by: Jane (Yuan) Xu <31798555+janeyx99@users.noreply.github.com>

2025-10-06 22:03:25 +00:00

..

_multi_tensor

[BE][PYFMT] migrate PYFMT for {torch,test}/{nn,optim}/** to ruff format (#144548)

2025-06-14 11:27:04 +00:00

__init__.py

[muon] Introduce Muon optimizer to PyTorch (#160213)

2025-08-24 08:03:04 +00:00

_adafactor.py

[Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735)

2025-07-08 18:10:33 +00:00

_functional.py

[BE][PYFMT] migrate PYFMT for {torch,test}/{nn,optim}/** to ruff format (#144548)

2025-06-14 11:27:04 +00:00

_muon.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

adadelta.py

[Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735)

2025-07-08 18:10:33 +00:00

adagrad.py

Document the rest of the specific optimizer module APIs (#158669)

2025-07-19 07:27:15 +00:00

adam.py

Make Adam, AdamW work with nonzero-dim Tensor betas (#149939)

2025-10-06 22:03:25 +00:00

adamax.py

[Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735)

2025-07-08 18:10:33 +00:00

adamw.py

Make Adam, AdamW work with nonzero-dim Tensor betas (#149939)

2025-10-06 22:03:25 +00:00

asgd.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

lbfgs.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

lr_scheduler.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

nadam.py

[Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735)

2025-07-08 18:10:33 +00:00

optimizer.py

Make Adam, AdamW work with nonzero-dim Tensor betas (#149939)

2025-10-06 22:03:25 +00:00

radam.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

rmsprop.py

[Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735)

2025-07-08 18:10:33 +00:00

rprop.py

[Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735)

2025-07-08 18:10:33 +00:00

sgd.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

sparse_adam.py

[Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735)

2025-07-08 18:10:33 +00:00

swa_utils.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00