VecEnv

public final class VecEnv<EnvType: Env, Element: TensorNumeric>
where EnvType.ActType == Tensor<Element>, EnvType.ObsType == Tensor<Element>
extension VecEnv: Env where EnvType.TerminatedType == Bool

  • Declaration

    Swift

    public init(count: Int, _ closure: (Int) throws -> EnvType) rethrows

Available where EnvType.TerminatedType == Bool

  • Declaration

    Swift

    public typealias ActType = EnvType.ActType

  • Declaration

    Swift

    public typealias ObsType = EnvType.ObsType

  • Declaration

    Swift

    public typealias RewardType = [EnvType.RewardType]

  • Declaration

    Swift

    public typealias TerminatedType = [Bool]

  • Declaration

    Swift

    public func step(action: ActType) -> (ObsType, RewardType, TerminatedType, [String : Any])

  • Declaration

    Swift

    public func reset(seed: Int?) -> (ObsType, [String : Any])

  • Declaration

    Swift

    public static var rewardThreshold: Float { get }

  • Declaration

    Swift

    public static var actionSpace: [ClosedRange<Float>] { get }

  • Declaration

    Swift

    public static var stateSize: Int { get }