Ejemplo n.º 1
0
def j_tensor_wrapper_to_np_array(j_obj: JavaObject):
    buffer, dtype_str, shape = j_obj.getBytes(), j_obj.getDtypeStr(
    ), j_obj.getShape()
    arr = np.frombuffer(buffer, dtype=dtype_str).reshape(shape)
    return arr