微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

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

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

像这样

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

解决方法

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

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。