numpy.float32をbytesに変換する方法

Question

前提・実現したいこと

Pytorchを用いたマルウェア検知のためのDeep Learningのモデルを作成しています．
その際にPytorchが扱えるように，bytes型で読み込んだマルウェアのバイナリーファイルを以下のコードを用いてTensor型に変換しています

BINARY = b'MZ\x90\x00\x03\x00\x00\x00\x04\x00' #f.read()で読み込んだ直後

inp_bytez = torch.from_numpy(np.frombuffer(BINARY, dtype=np.uint8)[np.newaxis,:]).float()
>> tensor([ 77.,  90., 144.,  ...,   0.,   0.,   0.])

そして色々な処理が終わった後に，要素の型がnumpy.float32のnumpy.ndarrayが返ってくるのですが，これを上記のBINARYのように変換する方法はありますでしょうか？
以下のようなコードを書いているのですが，想定していた形が得られません．

発生している問題

Return = [[90. 144.  0.  3.  0.  0.  0.  4.  0.  0.]] 
type(Return, Return[0],Return[0][0])
>>'numpy.ndarray', 'numpy.ndarray', 'numpy.float32'

bytez = list(map(lambda x: bytes(x), Return[0]))
#実際に出力された値
>> [b'\x00\x00\xb4B', b'\x00\x00\x10C', b'\x00\x00\x00\x00', b'\x00\x00@@', b'\x00\x00\x00\x00', b'\x00\x00\x00\x00', b'\x00\x00\x00\x00',b'\x00\x00\x80@', b'\x00\x00\x00\x00', b'\x00\x00\x00\x00']
#想定していた値
>>b'MZ\x90\x00\x03\x00\x00\x00\x04\x00'

Answer 1

astype(np.uint8)したあとtobytes()するとよいです。

import numpy as np

Return = np.array([[90., 144.,  0.,  3.,  0.,  0.,  0.,  4.,  0.,  0.]])
Return = Return.astype(np.float32)
print(type(Return[0][0])) # <class 'numpy.float32'>

ret = Return[0].astype(np.uint8)
print(ret) # [ 90 144   0   3   0   0   0   4   0   0]
ret = ret.tobytes()
print(ret) # b'Z\x90\x00\x03\x00\x00\x00\x04\x00\x00'

Comments

[退会済みユーザー]

できました！　ありがとうございます！！