Coreml:如何在切片上添加两个张量?

Coreml: How to add two tensors on slices?

我有两个张量:a.shape = [1, 3, 80, 80, 2] 和 b.shape = [1, 3, 80, 80, 19] 我只想添加 b[..., 0:2] + a 使用 coreml 模型构建器。

像这样

b[..., 0:2] = (b[..., 0:2] * 2. - 0.5 + a)

将 b 分成两部分,修改 0:2 切片,然后将其与 b 的另一个切片连接成一个 (1, 3, 80, 80, 19) 张量。