Create a tensor Args: shape (tuple or list): Tensor shape dtype (numpy dtype): Tensor data type (e.g. np.uint8) layout (string): Tensor layout (e.g. NC, HWC, NHWC) max_random (number or tuple or list): Maximum random value rng (numpy random Generator): To
(shape, dtype, layout, max_random=None, rng=None, transform_dist=None)
| 194 | |
| 195 | |
| 196 | def create_tensor(shape, dtype, layout, max_random=None, rng=None, transform_dist=None): |
| 197 | """Create a tensor |
| 198 | |
| 199 | Args: |
| 200 | shape (tuple or list): Tensor shape |
| 201 | dtype (numpy dtype): Tensor data type (e.g. np.uint8) |
| 202 | layout (string): Tensor layout (e.g. NC, HWC, NHWC) |
| 203 | max_random (number or tuple or list): Maximum random value |
| 204 | rng (numpy random Generator): To fill tensor with random values |
| 205 | transform_dist (function): To transform random values (e.g. dist_odd) |
| 206 | |
| 207 | Returns: |
| 208 | cvcuda.Tensor: The created tensor |
| 209 | """ |
| 210 | h_data = generate_data(shape, dtype, max_random, rng) |
| 211 | if transform_dist is not None: |
| 212 | vec_transform_dist = np.vectorize(transform_dist) |
| 213 | h_data = vec_transform_dist(h_data) |
| 214 | h_data = h_data.astype(dtype) |
| 215 | return to_cvcuda_tensor(h_data, layout) |
| 216 | |
| 217 | |
| 218 | def to_cvcuda_image(host_data): |
nothing calls this directly
no test coverage detected