fn
add_storages
→Storageint add_storages(const int & a, const int & b, int numel, Dtype dt, Device device)Element-wise addition: returns a + b.
Parameters
a, bconst Storage&Input buffers.
numelstd::size_tNumber of elements.
dtDtypeElement dtype.
deviceDeviceTarget device.
Returns
StorageNew storage holding the element-wise sum.