torch.TensorOptions

export interface TensorOptions<D extends DType = DType, Dev extends DeviceType = DeviceType> {
  dtype?: D;
  device?: Dev;
  requires_grad?: boolean;
}

Dextends DType

Devextends DeviceType

dtype(D)optional
device(Dev)optional
requires_grad(boolean)optional

Options for tensor creation with compile-time dtype and device tracking.

Examples

const a = torch.zeros(2, 3, { dtype: 'int32' });  // Tensor<[2, 3], 'int32'>
const b = torch.zeros(2, 3, { device: 'cpu' });   // Tensor<[2, 3], 'float32', 'cpu'>