当前位置:网站首页>【mindspore】【Categorical】softmax数据放入Categorical类出现和不为1的错误
【mindspore】【Categorical】softmax数据放入Categorical类出现和不为1的错误
2022-08-10 03:28:00 【小乐快乐】
Categorical类
import mindspore
import mindspore.nn as nn
import numpy as np
from mindspore import Tensor
from mindspore.nn.probability.distribution import Categorical
class Actor(nn.Cell):
def __init__(self, state_dim, action_dim):
super().__init__()
self.fc = nn.SequentialCell([
nn.Dense(in_channels=state_dim, out_channels=32),
nn.ReLU(),
nn.Dense(in_channels=32, out_channels=action_dim),
nn.Tanh()
])
self.softmax = nn.Softmax(axis=-1)
def construct(self, state):
out = self.fc(state)
return self.softmax(out)
actor = Actor(4, 2)
for i in range(1000):
x = Tensor(np.random.random(size=(10, 4)),dtype=mindspore.float32)
y = actor(x)
print(y)
cat = Categorical(y)
index = cat.sample()
print(index)
【操作步骤&问题现象】
今天使用了nn.softmax函数对神经网络输出处理,经过对神经网络的输出处理后,变为一个相加为1的Tensor,然后我用这个Tensor去输入到Categorical中,在训练过程中,有时会出现数据放到Categorical里面会出现相加不为1的错误
对于同样的数据,我把它放到pytorch的Categorical类里面处理发现没有错误,由此我在想是否是因为精度的原因,比如X为此tensor,允许X.sum() = 1 - epsilon,这个epsilon是否可以调大一点?
mindspore1.5-cpu,python3.7.5,windows 10环境下,Categorical类中的精度调整之后发现就不报错了。
边栏推荐
猜你喜欢
随机推荐
【Verilog数字系统设计(夏雨闻)5-------模块的结构、数据类型、变量和基本运算符号1】
program internationalization
云程发轫,万里可期 | 云扩科技再次入选Gartner《2022年中国ICT技术成熟度曲线报告》
驱动程序开发:按键中断之异步通知
TCP协议之《数据与控制流程交叉时的延迟处理》
小程序wxs
一文教会你快速上手 Vim
ARP Spoofing - Tutorial Details
如何开启热部署Devtools
学习总结week4_2正则
matlab simulink response spectrum calculation
超全面的Android面试题汇总
请教各位confluence部署连接数据库成功,但是在后面建表设置的时候报错
嵌入式分享合集32
mediaserver创建
笔试题记录
C语言顺序表(源码)
如何将 @Transactional 事务注解运用到炉火纯青?
测试工作管理与规范
Evaluation and Construction of Enterprise Network Security Capability from the Sliding Ruler Model