tf.minimum()用法

本文详细介绍了TensorFlow中的tf.math.minimum函数,该函数用于计算两个Tensor元素级的最小值,要求输入的Tensor数据类型一致。适用于深度学习和机器学习领域的数据处理和算法实现。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

import tensorflow as tf import numpy as np import matplotlib.pyplot as plt from carModel import CarEnvironment print("TensorFlow 版本:", tf.__version__) print("GPU 可用性:", tf.config.list_physical_devices('GPU')) # === 超参数 === EP_MAX = 1000 # 训练回合数 EP_LEN = 1800 # 每回合步长(与工况数据长度一致) GAMMA = 0.95 # 折扣因子 A_LR = 0.0001 # Actor学习率 C_LR = 0.0002 # Critic学习率 BATCH_SIZE = 128 # 训练批量大小 UPDATE_STEPS = 10 # 训练时,PPO的更新步数 EPSILON = 0.2 # PPO的裁剪系数 HIDDEN_UNITS = 128 # 隐藏层单元数 env = CarEnvironment() S_DIM = env.observation_space.shape[0] # 状态空间维度 A_DIM = env.action_space.shape[0] # 动作空间维度 A_BOUND = env.action_space.high # 动作边界 class PPOAgent: def __init__(self): # 配置GPU参数 config = tf.ConfigProto() config.gpu_options.allow_growth = True # 动态申请显存 config.gpu_options.visible_device_list = "0" # 使用第一个GPU self.sess = tf.Session(config=config) self._build_model() self.sess.run(tf.global_variables_initializer()) def _build_model(self): """ 建立 PPO 网络 """ # 1️⃣ **输入层** self.S = tf.placeholder(tf.float32, [None, S_DIM], 'state') self.A = tf.placeholder(tf.float32, [None, A_DIM], 'action') self.R = tf.placeholder(tf.float32, [None, 1], 'reward') self.advantage = tf.placeholder(tf.float32, [None, 1], 'advantage') # 2️⃣ **Critic(评估网络)** with tf.variable_scope('Critic'): c_hidden = tf.layers.dense(self.S, HIDDEN_UNITS, tf.nn.relu) self.v = tf.layers.dense(c_hidden, 1) self.critic_loss = tf.reduce_mean(tf.square(self.R - self.v)) self.critic_train_op = tf.train.AdamOptimizer(C_LR).minimize(self.critic_loss) # 3️⃣ **Actor(策略网络)** pi, pi_params = self._build_actor('pi', trainable=True) # 当前策略 oldpi, oldpi_params = self._build_actor('oldpi', trainable=False) # 旧策略
最新发布
03-12
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值