如何解决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 举报,一经查实,本站将立刻删除。