mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
548c460bf1
See https://github.com/pytorch/pytorch/pull/129751#issue-2380881501. Most changes are auto-generated by linter. You can review these PRs via: ```bash git diff --ignore-all-space --ignore-blank-lines HEAD~1 ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/129758 Approved by: https://github.com/ezyang
11 lines
302 B
Python
11 lines
302 B
Python
import torch
|
|
|
|
|
|
input = []
|
|
input.append(torch.tensor([1.0, 2.0, 3.0, 4.0]))
|
|
input.append(torch.tensor([[1.0, 2.0, 3.0, 4.0]]))
|
|
input.append(torch.tensor([[[1.0, 2.0, 3.0, 4.0]]]))
|
|
reveal_type(input[0].shape[0]) # E: int
|
|
reveal_type(input[1].shape[1]) # E: int
|
|
reveal_type(input[2].shape[2]) # E: int
|