• 文档 >
  • 结合输入操作符
快捷方式

结合输入操作符

std::tuple<at::Tensor, at::Tensor, at::Tensor> tbe_input_combine_cpu(const std::vector<at::Tensor> &indices_list, const std::vector<at::Tensor> &offsets_list, const std::vector<at::Tensor> &per_sample_weights, const at::Tensor &include_last_offsets)
std::tuple<at::Tensor, at::Tensor, at::Tensor> padding_fused_tbe_input_combine_cpu(const std::vector<at::Tensor> &indices_list, const std::vector<at::Tensor> &offsets_list, const std::vector<at::Tensor> &per_sample_weights, const at::Tensor &include_last_offsets, int64_t batch_size)

文档

访问 PyTorch 的综合开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获取问题解答

查看资源