如何解决更改 MNIST 张量内的值
出于实验原因,我试图更改大小为 (64,1,28,28) 的张量内的值,该张量包含来自 MNIST 数据集的 64 个图像的像素值。我正在尝试将它们全部替换为一个。
然而,我试图应用 where
的 torch 方法无济于事。
main = torch.randn(64,28)
ones = torch.ones(64,28)
torch.where(main < 0,ones,ones)
torch.where(main > 0,ones)
where
方法是否仅适用于较小的维度?
如果是这样,如何以更好的方式做到这一点?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。