File: tensor.go

package info (click to toggle)
golang-github-nlpodyssey-gopickle 0.3.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 732 kB
  • sloc: python: 32; makefile: 5
file content (13 lines) | stat: -rw-r--r-- 318 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
 
// Copyright 2020 NLP Odyssey Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package pytorch

type Tensor struct {
	Source        StorageInterface
	StorageOffset int
	Size          []int
	Stride        []int
	RequiresGrad  bool
}