import numpy as np
def discretize_action(a, n_bins=256, lo=-1.0, hi=1.0):
bins = np.linspace(lo, hi, n_bins)
return np.digitize(a, bins)
action = np.array([0.42, -0.87, 0.05, 0.0, 0.0, 0.33, 1.0]) # 7-DoF continuous action
tokens = discretize_action(action)
print("continuous action:", action)
print("discretized tokens:", tokens)continuous action: [ 0.42 -0.87 0.05 0. 0. 0.33 1. ]
discretized tokens: [182 17 134 128 128 170 256]