torch.spilt

`torch.split`是PyTorch库中的一个函数，它用于将张量分割成多个小张量（sub-tensors）。这个操作通常在处理需要切分数据集、模型输入等场景时很有用。它接受三个参数：

1. `tensor`: 需要被分割的原始张量。
2. `split_size_or_sections`: 这可以是一个整数，表示每个分割出来的部分应该大致的大小；也可以是一个列表或元组，指定每一个分割的部分长度。
3. `dim` (可选): 默认值是0，表示沿着第一个维度（行或列，取决于张量的形状）进行分割。

返回的是一个包含分割后张量的tuple，这些张量按照提供的分割策略排列。

例如：

import torch

# 假设我们有一个形状为(64, 10)的张量
input_tensor = torch.randn(64, 10)

# 分割为两部分，每部分大小为32和32
split_tensors = torch.split(input_tensor, [32, 32], dim=0)