您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
36 行
2.0 KiB
36 行
2.0 KiB
namespace Unity.MLAgents.Inference
|
|
{
|
|
/// <summary>
|
|
/// Contains the names of the input and output tensors for the Inference Brain.
|
|
/// </summary>
|
|
internal static class TensorNames
|
|
{
|
|
public const string BatchSizePlaceholder = "batch_size";
|
|
public const string SequenceLengthPlaceholder = "sequence_length";
|
|
public const string VectorObservationPlaceholder = "vector_observation";
|
|
public const string RecurrentInPlaceholder = "recurrent_in";
|
|
public const string recurrentInPlaceholderH = "recurrent_in_h";
|
|
public const string recurrentInPlaceholderC = "recurrent_in_c";
|
|
public const string VisualObservationPlaceholderPrefix = "visual_observation_";
|
|
public const string ObservationPlaceholderPrefix = "obs_";
|
|
public const string PreviousActionPlaceholder = "prev_action";
|
|
public const string ActionMaskPlaceholder = "action_masks";
|
|
public const string RandomNormalEpsilonPlaceholder = "epsilon";
|
|
|
|
public const string ValueEstimateOutput = "value_estimate";
|
|
public const string RecurrentOutput = "recurrent_out";
|
|
public const string recurrentOutputH = "recurrent_out_h";
|
|
public const string recurrentOutputC = "recurrent_out_c";
|
|
public const string MemorySize = "memory_size";
|
|
public const string VersionNumber = "version_number";
|
|
public const string ContinuousActionOutputShape = "continuous_action_output_shape";
|
|
public const string DiscreteActionOutputShape = "discrete_action_output_shape";
|
|
public const string ContinuousActionOutput = "continuous_actions";
|
|
public const string DiscreteActionOutput = "discrete_actions";
|
|
|
|
// Deprecated TensorNames entries for backward compatibility
|
|
public const string IsContinuousControlDeprecated = "is_continuous_control";
|
|
public const string ActionOutputDeprecated = "action";
|
|
public const string ActionOutputShapeDeprecated = "action_output_shape";
|
|
}
|
|
}
|