fn
multiply_storages
→Storageint multiply_storages(const int & a, const int & b, int numel, Dtype dt, Device device)Element-wise multiplication: returns a * b.
Parameters
a, bconst Storage&Input buffers (must match
numel and dt).numelstd::size_tNumber of elements.
dtDtypeElement dtype.
deviceDeviceTarget device.
Returns
StorageNew storage holding the element-wise product.