U <ºcXã@s0dZddlZddlZdd„Zdd„Zdd„ZdS) z0 This module converts objects into numpy array. éNcCsbt|tjƒr|St|tƒr"t|ƒSt |¡r8t |g¡St|tjƒrLt |ƒSt d  t |ƒ¡ƒ‚dS)zv Args: x: An instance of torch tensor or caffe blob name Returns: numpy.array: Numpy array zHGot {}, but numpy array, torch tensor, or caffe2 blob name are expected.N) Ú isinstanceÚnpZndarrayÚstrÚ_prepare_caffe2ZisscalarÚarrayÚtorchZTensorÚ_prepare_pytorchÚNotImplementedErrorÚformatÚtype©Úx©rúG/tmp/pip-unpacked-wheel-gikjz4vx/torch/utils/tensorboard/_convert_np.pyÚmake_nps     ÿÿrcCs| ¡ ¡ ¡}|S)N)ÚdetachÚcpuÚnumpyr rrrrsrcCsddlm}| |¡}|S)Nr)Ú workspace)Z caffe2.pythonrZ FetchBlob)r rrrrr$s  r)Ú__doc__rrrrrrrrrrÚs