fn
square_storage
→Storageint square_storage(const int & s, int numel, Dtype dt, Device device)Element-wise square: returns s * s.
Parameters
sconst Storage&Input buffer.
numelstd::size_tNumber of elements.
dtDtypeElement dtype.
deviceDeviceTarget device.
Returns
StorageNew storage holding the element-wise square.