def shape_list(self):
tensor_shape_list = []
num_axes = oneflow_api.OfBlob_NumAxes(self.of_blob_ptr_)
oneflow_api.OfBlob_ResetTensorIterator(self.of_blob_ptr_)
while not oneflow_api.OfBlob_CurTensorIteratorEqEnd(self.of_blob_ptr_):
shape_tensor = np.zeros(self.num_axes, dtype=np.int64)
oneflow_api.OfBlob_CurTensorCopyShapeTo(self.of_blob_ptr_, shape_tensor)
assert len(shape_tensor.shape) == 1
assert shape_tensor.size == num_axes
tensor_shape_list.append(tuple(shape_tensor.tolist()))
oneflow_api.OfBlob_IncTensorIterator(self.of_blob_ptr_)
return tensor_shape_list
def num_axes(self):
return oneflow_api.OfBlob_NumAxes(self.of_blob_ptr_)
def set_shape(self, shape):
assert isinstance(shape, (list, tuple))
assert len(shape) == oneflow_api.OfBlob_NumAxes(self.of_blob_ptr_)
oneflow_api.OfBlob_CopyShapeFromNumpy(
self.of_blob_ptr_, np.array(shape, dtype=np.int64)
)
def shape(self):
num_axes = oneflow_api.OfBlob_NumAxes(self.of_blob_ptr_)
dst_ndarray = np.zeros(num_axes, dtype=np.int64)
oneflow_api.OfBlob_CopyShapeToNumpy(self.of_blob_ptr_, dst_ndarray)
return tuple(dst_ndarray.tolist())
def static_shape(self):
num_axes = oneflow_api.OfBlob_NumAxes(self.of_blob_ptr_)
dst_ndarray = np.ndarray(num_axes, dtype=np.int64)
oneflow_api.OfBlob_CopyStaticShapeTo(self.of_blob_ptr_, dst_ndarray)
return tuple(dst_ndarray.tolist())