摘要: torch.repeat_interleave 用于按指定规则重复张量的元素,支持按维度扩展或自定义每个元素的重复次数。以下是详细说明和示例: 作用 功能:沿特定维度重复张量的元素,支持两种模式: 统一重复次数:所有元素重复相同次数。 自定义重复次数:每个元素按单独指定的次数重复。 与 torch. 阅读全文
posted @ 2025-02-22 15:54 最爱丁珰 阅读(3) 评论(0) 推荐(0) 编辑
点击右上角即可分享
微信分享提示