fn
reciprocal_storage
→Storageint reciprocal_storage(const int & s, int numel, Dtype dt, Device device)Element-wise reciprocal: returns 1 / s.
Parameters
sconst Storage&Input buffer (non-zero).
numelstd::size_tNumber of elements.
dtDtypeElement dtype.
deviceDeviceTarget device.
Returns
StorageNew storage holding 1 / s.