YouTip

# Returns the median of all elements print("Global Median:", torch.median(x)) # Along dim=0 Median y = torch.tensor([[1,3,2],[4,1,3]]) print("dim=0 Median:", torch.median(y, dim=0))

The output result is:

Global Median: tensor(3) dim=0 Median: torch.return_types.median(values=tensor([1, 1, 2]), indices=tensor([0, 1, 0]))

---

Pytorch torch Reference Manual

AI is thinking...

PyTorch Data Transforms

PyTorch torch.nn Reference Manual

ByteArk Coding Plan supports mainstream large models like Doubao, GLM, DeepSeek, Kimi, MiniMax, etc., official supply stable and reliable. Configuration Guide ¥9.9/ month Open Now

iFlytek Xingchen Coding Plan includes free model call quotas, DeepSeek, GLM, Kimi, MiniMax, a one-stop experience and deployment platform. Configuration Guide ¥3.9/ month Open Now

Click to Share Notes

Cancel

Write notes...

Image URL

Image Description

Image Size ×

Share Notes

Nickname