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) 张量。
我有两个张量: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) 张量。