如何解决如何制作 2D softmax 层?
我正在使用 keras 库制作网络。
假设我有二维矩阵
[ 0 0 1 2
0 1 2 5
1 0 0 1 ]
我想要做的是获得以下矩阵
[ 0.00 0.00 0.02 0.10
0.00 0.02 0.10 0.99
0.02 0.00 0.00 0.02 ]
如图所示,我想让图层只表示要强调的二维数组的最大元素。
我怎样才能做到这一点?
这是否可以通过简单地将 softmax 调整两次来实现?
解决方法
您无需担心 2d 形状,softmax 可以正常工作。
cmd /c "docker exec -t my-container pg_dump my_db -c -U postgres | gzip > C:\\backup\\my_db.gz"
import tensorflow as tf
inputs = tf.random.normal(shape=(3,3))
outputs = tf.keras.activations.softmax(inputs)
print(inputs)
print(outputs)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。