Add 'scalar_type='