fn
subtract_storages
→Storageint subtract_storages(const int & a, const int & b, int numel, Dtype dt, Device device)Element-wise subtraction: 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 difference.