ExTorch.Tensor.Options (extorch v0.1.0-pre0)

The ExTorch.Tensor.Options struct defines the creation parameters of a tensor.

Link to this section Summary

Types

t()

An ExTorch.Tensor.Options defines the creation parameters of a tensor.

Link to this section Types

@type t() :: %ExTorch.Tensor.Options{
  device: ExTorch.Device.device(),
  dtype: ExTorch.DType.dtype(),
  layout: ExTorch.Layout.layout(),
  memory_format: ExTorch.MemoryFormat.memory_format(),
  pin_memory: boolean(),
  requires_grad: boolean()
}

An ExTorch.Tensor.Options defines the creation parameters of a tensor.