Table of Contents

社区[ - ][ + ]

PyTorch 治理 | 构建 + CI
PyTorch 贡献指南
PyTorch 设计哲学
PyTorch 治理 | 机制
PyTorch 治理 | 维护者

开发者笔记[ - ][ + ]

CUDA 自动混合精度示例
自动求导机制
广播语义
CPU 线程与 TorchScript 推理
CUDA 语义
分布式数据并行
扩展 PyTorch
使用 autograd.Function 扩展 torch.func
常见问题
FSDP 笔记
梯度检查机制
HIP (ROCm) 语义
大规模部署的功能
模块
MPS 后端
多进程最佳实践
数值精度
可重复性
序列化语义
Windows 常见问题

语言绑定[ - ][ + ]

C++
Javadoc
torch::deploy

Python API[ - ][ + ]

库[ - ][ + ]

torchaudio
TorchData
TorchRec
TorchServe
torchtext
torchvision
PyTorch on XLA Devices

Docs >
torch.Tensor >
torch.Tensor.split

Shortcuts

torch.Tensor.split¶

Tensor.split(split_size, dim=0)[源代码]¶: 参见 torch.split()

版权所有 2023，PyTorch 贡献者。

Built with Sphinx using a theme provided by Read the Docs.

torch.Tensor.split
- Tensor.split()