|
|
|
@ -2030,7 +2030,7 @@ class OneHot(PrimitiveWithInfer):
|
|
|
|
|
depth_val = depth['value']
|
|
|
|
|
validator.check_integer("depth", depth_val, 0, Rel.GE, self.name)
|
|
|
|
|
# create new dimension at end if self.axis is -1
|
|
|
|
|
indices_shp.insert(self.axis, depth_val) if self.axis >= 0 else indices_shp.append(depth_val)
|
|
|
|
|
_ = indices_shp.insert(self.axis, depth_val) if self.axis >= 0 else indices_shp.append(depth_val)
|
|
|
|
|
|
|
|
|
|
return {'shape': indices_shp,
|
|
|
|
|
'dtype': on_value['dtype'],
|
|
|
|
|